hdu 3401 Trade 单调队列优化dp - c++编程基础

题目意思：

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

要求：

每天买卖的股票数量有限制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)股时为dp[i-w-1][k]+(k-j)*s[i] j=

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

代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#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=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;
}