题目链接
http://www.lydsy.com/JudgeOnline/problem.php?id=1855
思路
很显然是个DP题,比较容易想到下面的DP做法:
用
f[i][j]
表示第
i
天,手上有
而DP方程则为(以转移情况为购入股票举例)
这样做的正确性是显然的, f[i][j] 一定是前 i 天里手持
变化一下这个DP方程:
一般可以用单调队列优化的DP都能满足形如
f[x]=max(min){f[k]}+g[x]
的形式,那么我们可以把这里的
f[i−w−1][k]
看作是上面的
f[k]
,
apj
看作是上面的
g[x]
,于是就可以从小到大枚举
j
,并用单调队列维护二元组
而卖出的做法也非常相似,在这里就不再赘述了
代码
刚开始数组开大了,直接卡成TLE。。。郁闷。。。
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
#define MAXN 2100
#define INF 0x3f3f3f3f
using namespace std;
pair<int,int>q[MAXN];
int f[MAXN][MAXN];
int T,maxP,w;
int main()
{
int ans=-INF;
memset(f,-INF,sizeof(f));
scanf("%d%d%d",&T,&maxP,&w);
for(int i=1;i<=T;i++)
{
int ap,bp,as,bs;
scanf("%d%d%d%d",&ap,&bp,&as,&bs);
for(int j=0;j<=as;j++) f[i][j]=-ap*j; //初始化f[i][j]=-ap*j(光买了股票)
for(int j=0;j<=maxP;j++) f[i][j]=max(f[i][j],f[i-1][j]); //初始化什么都没买的情况更新f数组
int k=i-w-1; //单调队列里保存的就是(j,f[w][j]+ap[i]*j(-bp[i]*j))这个二元组
if(k>=0)
{
int h=0,t=0;
for(int j=0;j<=maxP;j++)
{
//买入股票
while(h<t&&q[h].first<j-as) h++;
while(h<t&&q[t-1].second<=f[k][j]+ap*j) t--;
q[t++]=make_pair(j,f[k][j]+ap*j);
if(h<t) f[i][j]=max(f[i][j],q[h].second-ap*j);
}
h=0,t=0;
for(int j=maxP;j>=0;j--)
{
//卖出股票
while(h<t&&q[h].first>j+bs) h++; //!!!!!
while(h<t&&q[t-1].second<=f[k][j]+bp*j) t--;
q[t++]=make_pair(j,f[k][j]+bp*j);
if(h<t) f[i][j]=max(f[i][j],q[h].second-bp*j);
}
}
ans=max(ans,f[i][0]); //显然以某天交易结束后收手不干,肯定是手上没有股票是最优的
}
printf("%d\n",ans);
return 0;
}