首页 > 代码库 > hdu-3401-Trade-单调队列优化的DP
hdu-3401-Trade-单调队列优化的DP
单调队列入门题。。。
dp[i][j]:第i天。手中拥有j个股票时,获得的最大利润。
若第i天不买不卖:dp[i][j]=max(dp[i][j],dp[i-1][j]);
若第i天买 :dp[i][j]=max(dp[i][j],dp[i-w-1][k]-(j-k)*ap[i]);
若第i天卖 :dp[i][j]=max(dp[i][j],dp[i-w-1][k]+(k-j)*bp[i]);
若仅仅考虑买的情况:
dp[i][j]=dp[i-w-1][k]+ap[i]*k-j*ap[i];
非常明显。能够用单调队列优化dp[i-w-1][k]+ap[i]*k。
#include<stdio.h> #include<string.h> #include<algorithm> #include<iostream> #include<queue> using namespace std; //#define INF ((1<<30)-1) #define INF 0xfffff #define maxn 2200 #define LL long long #define MOD 1000000009 int dp[2200][2200]; struct list { int val; int x; } p[5001],q; int ap[maxn],bp[maxn],as[maxn],bs[maxn]; int main() { int Ts; int n,m,w,i,j; scanf("%d",&Ts); while(Ts--) { scanf("%d%d%d",&n,&m,&w); for(i=0; i<=n; i++) for(j=0; j<=m; j++)dp[i][j]=-INF; int head,tail; for(i=1;i<=n;i++)scanf("%d%d%d%d",&ap[i],&bp[i],&as[i],&bs[i]); for(j=1;j<=w+1;j++) for(i=0;i<=min(as[j],m);i++) dp[j][i]=-1*ap[j]*i; for(i=2; i<=n; i++) { for(j=0;j<=m;j++)dp[i][j]=max(dp[i-1][j],dp[i][j]); if(i<=w+1)continue; head=1; tail=0; for(j=0; j<=m; j++) { q.x=j; q.val=dp[i-w-1][j]+ap[i]*j; while(tail>=head&&q.val>p[tail].val)tail--; p[++tail]=q; while(tail>=head&&p[head].x<j-as[i])head++; if(head<=tail)dp[i][j]=max(dp[i][j],p[head].val-ap[i]*j); } head=1; tail=0; for(j=m;j>=0;j--) { q.x=j; q.val=dp[i-w-1][j]+bp[i]*j; while(tail>=head&&q.val>p[tail].val)tail--; p[++tail]=q; while(tail>=head&&p[head].x>j+bs[i])head++; if(head<=tail)dp[i][j]=max(dp[i][j],p[head].val-bp[i]*j); } } int maxx=0; for(i=0;i<=m;i++)maxx=max(maxx,dp[n][i]); printf("%d\n",maxx); } return 0; }
hdu-3401-Trade-单调队列优化的DP
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。