Description
最近lxhgww又迷上了投资股票,通过一段时间的观察和学习,他总结出了股票行情的一些规律。 通过一段时间的观察,lxhgww预测到了未来T天内某只股票的走势,第i天的股票买入价为每股APi,第i天的股票卖出价为每股BPi(数据保证对于每个i,都有APi>=BPi),但是每天不能无限制地交易,于是股票交易所规定第i天的一次买入至多只能购买ASi股,一次卖出至多只能卖出BSi股。 另外,股票交易所还制定了两个规定。为了避免大家疯狂交易,股票交易所规定在两次交易(某一天的买入或者卖出均算是一次交易)之间,至少要间隔W天,也就是说如果在第i天发生了交易,那么从第i+1天到第i+W天,均不能发生交易。同时,为了避免垄断,股票交易所还规定在任何时间,一个人的手里的股票数不能超过MaxP。 在第1天之前,lxhgww手里有一大笔钱(可以认为钱的数目无限),但是没有任何股票,当然,T天以后,lxhgww想要赚到最多的钱,聪明的程序员们,你们能帮助他吗?
【题目分析】
单调队列优化DP
【代码】
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int t,maxp,w;
int a[2001],b[2001],sa[2001],sb[2001];
int dp[2001][2001];
int que[2001],hd=1,tl=0;
inline int read()
{
int ret=0,f=1;
char ch=getchar();
while (ch<'0'||ch>'9')
{
if (ch=='-') f=-1;
ch=getchar();
}
while (ch>='0'&&ch<='9')
{
ret*=10;
ret+=ch-'0';
ch=getchar();
}
return ret*f;
}
//int main()
//{
// t=read();maxp=read();w=read();
// for (int i=1;i<=t;++i)
// a[i]=read(),b[i]=read(),sa[i]=read(),sb[i]=read();
// memset(dp,-0x3f,sizeof dp);
// dp[0][0]=0;
// for (int i=1;i<=t;++i)
// {
// for (int j=0;j<=sa[i];++j) dp[i][j]=max(dp[i][j],-a[i]*j);
// for (int j=0;j<=maxp;++j)
// {
// dp[i][j]=max(dp[i-1][j],dp[i][j]);
// if (i-w-1>=0) for (int k=max(j-sa[i],0);k<j;++k)
// dp[i][j]=max(dp[i][j],dp[i-w-1][k]-(j-k)*a[i]);
// if (i-w-1>=0)for (int k=j+1;k<=min(j+sb[i],maxp);++k)
// dp[i][j]=max(dp[i][j],dp[i-w-1][k]+(k-j)*b[i]);
// }
// }
// printf("%d\n",dp[t][0]);
//}
int main()
{
t=read();maxp=read();w=read();
for (int i=1;i<=t;++i)
a[i]=read(),b[i]=read(),sa[i]=read(),sb[i]=read();
memset(dp,-0x3f,sizeof dp);
dp[0][0]=0;
for (int i=1;i<=t;++i)
{
for (int j=0;j<=sa[i];++j) dp[i][j]=max(dp[i][j],-a[i]*j);
for (int j=0;j<=maxp;++j) dp[i][j]=max(dp[i-1][j],dp[i][j]);
if (i-w-1>=0)
{
hd=1;tl=0;
for (int j=0;j<=maxp;++j)
{
while (hd<=tl&&j>sa[i]+que[hd]) hd++;
while (hd<=tl&&dp[i-w-1][j]+j*a[i]>=dp[i-w-1][que[tl]]+que[tl]*a[i]) tl--;
que[++tl]=j;
dp[i][j]=max(dp[i][j],dp[i-w-1][que[hd]]-(j-que[hd])*a[i]);
}
hd=1;tl=0;
for (int j=maxp;j>=0;--j)
{
while (hd<=tl&&que[hd]>min(maxp,j+sb[i])) hd++;
while (hd<=tl&&dp[i-w-1][j]+j*b[i]>=dp[i-w-1][que[tl]]+que[tl]*b[i]) tl--;
que[++tl]=j;
dp[i][j]=max(dp[i][j],dp[i-w-1][que[hd]]+(que[hd]-j)*b[i]);
}
}
}
printf("%d\n",dp[t][0]);
}