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)|1
using 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
",ans);

   }
   return 0;
}