hdu 3401 Trade 单调队列优化dp

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=3401

题目意思：

一个人开始有无穷多的钱，问买股票最多能赚多少钱。

要求：

每天买卖的股票数量有限制bl和sl，交易的时间间隔必须超过w+1天。

最多持有的数量为p股。

解题思路：

dp[i][j]表示前i天当持有j股股票时，获得的最大利益。

状态转移：

当第i天不交易时为dp[i-1][j];

当第i天买(j-k)股时为dp[i-w-1][k]-(j-k)*b[i]   0=<k<=j  //注意要求第i-w----i-1天都不能交易.

     卖(k-j)股时为dp[i-w-1][k]+(k-j)*s[i]     j=<k<=p

当固定某个j时，dp[i-w-1][k]+k*b[i] 是固定的(k,j有个关系），所以可以用单调队列优化。

代码:

#include<iostream>#include<cmath>#include<cstdio>#include<cstdlib>#include<string>#include<cstring>#include<algorithm>#include<vector>#include<map>#include<stack>#include<list>#include<queue>#define eps 1e-6#define INF 0x1f1f1f1f#define PI acos(-1.0)#define ll __int64#define lson l,m,(rt<<1)#define rson m+1,r,(rt<<1)|1using namespace std;/*freopen("data.in","r",stdin);freopen("data.out","w",stdout);*/int dp[2200][2200],b[2200],s[2200],bl[2200],sl[2200];struct Node{   int nu,mo;}myd[2200];int main(){   int w,t,p,ca;   scanf("%d",&ca);   while(ca--)   {      scanf("%d%d%d",&t,&p,&w);      for(int i=1;i<=t;i++)         scanf("%d%d%d%d",&b[i],&s[i],&bl[i],&sl[i]);      memset(dp,-INF,sizeof(dp));      for(int j=1;j<=t;j++) //所有股票都是当天买的         for(int i=0;i<=min(bl[j],p);i++)            dp[j][i]=max(dp[j][i],-i*b[j]);      for(int i=2;i<=t;i++)      {         dp[i][0]=0;         for(int j=0;j<=p;j++)   //没有交易            dp[i][j]=max(dp[i-1][j],dp[i][j]);        if(i<w+2)  //不能交易            continue;        int pre=i-w-1;        int ss=0,e=-1;        for(int j=0;j<=p;j++) //买        {           int tmp=dp[pre][j]+j*b[i];            while(ss<=e&&tmp>=myd[e].mo)                e--;           struct Node te;           te.nu=j,te.mo=tmp;           myd[++e]=te;           while(ss<=e&&j-myd[ss].nu>bl[i])               ss++;           dp[i][j]=max(dp[i][j],myd[ss].mo-j*b[i]);        }        ss=0,e=-1;       // memset(myd,0,sizeof(myd));        for(int j=p;j>=0;j--) //卖        {            int tmp=dp[pre][j]+j*s[i];            while(ss<=e&&tmp>=myd[e].mo)                e--;            struct Node te;            te.nu=j;te.mo=tmp;            myd[++e]=te;            while(ss<=e&&myd[ss].nu-j>sl[i])                ss++;            dp[i][j]=max(dp[i][j],myd[ss].mo-j*s[i]);        }      }      int ans=0;      for(int i=0;i<=p;i++)        ans=max(ans,dp[t][i]);      printf("%d\n",ans);   }   return 0;}