hdu3401 Trade（单调队列优化dp）

hdu3401

题目

知道之后n天的股票买卖价格（api，bpi），以及每天股票买卖数量上限（asi，bsi），问他最多能赚多少钱。开始时有无限本金，要求任两次交易需要间隔W天以上，即第i天交易，第i+w+1天才能再交易。同时他任意时刻最多只能拥有maxp的股票

思路

单调队列优化的例题。
http://www.cnblogs.com/ka200812/archive/2012/07/11/2585950.html

代码

#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
#include <iostream>
#include <stack>
#include <queue>

using namespace std;

const int maxn=2010;

struct node
{
    int val;
    int num;
    node(int v=0,int n=0)
    {
        val=v;
        num=n;
    }
};
deque<node> q;

int t,maxp,w;
int ap[maxn],bp[maxn],as[maxn],bs[maxn];
int dp[maxn][maxn];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&t,&maxp,&w);
        for(int i=1; i<=t; i++)
            scanf("%d%d%d%d",&ap[i],&bp[i],&as[i],&bs[i]);
        for(int i=0; i<=t; i++)
            for(int j=0; j<=maxp; j++)
                dp[i][j]=-0x3f3f3f3f;
        for(int i=1; i<=w+1; i++)
            for(int j=0; j<=min(maxp,as[i]); j++)
                dp[i][j]=-ap[i]*j;
        dp[0][0]=0;
        for(int i=1; i<=t; i++)
        {
            for(int j=0; j<=maxp; j++)
                dp[i][j]=max(dp[i][j],dp[i-1][j]);

            if(i<=w+1) continue;
            q.clear();
            int pre=i-w-1;
            for(int j=0; j<=maxp; j++)
            {
                int temp=dp[pre][j]+j*ap[i];
                while(!q.empty()&&q.back().num<temp)
                    q.pop_back();
                q.push_back(node(j,temp));
                while(!q.empty()&&j-q.front().val>as[i])
                    q.pop_front();
                dp[i][j]=max(dp[i][j],q.front().num-j*ap[i]);
            }
            q.clear();
            for(int j=maxp; j>=0; j--)
            {
                int temp=dp[pre][j]+j*bp[i];
                while(!q.empty()&&q.back().num<temp)
                    q.pop_back();
                q.push_back(node(j,temp));
                while(!q.empty()&&q.front().val-j>bs[i])
                    q.pop_front();
                dp[i][j]=max(dp[i][j],q.front().num-j*bp[i]);
            }
        }
        int ans = 0;
        for(int i = 0; i <= maxp; i++)
        {
            ans = max(ans,dp[t][i]);
        }
        printf("%d\n",ans);
    }
    return 0;
}