[bzoj 1855] 股票交易

题意:

最近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+单调队列优化。

设dp[i][j]为决策到第i天手上拥有j股的最大收益。

刚开始复杂度是\(O(n^2×m^2)\)的。

我们每次将dp[i][j]与dp[i-1][j]取max,这是不影响答案的,这样我们就可以让dp[i][j]直接被dp[max(0,i-w-1)][j]转移,复杂度为\(O(n×m^2)\)

设k=max(0,i-w-1)接下来就是第i天的转移:

\(dp[i][j]=\max{dp[k][y]-ap[i]×(j-y)} (j-y>=0)\)
\(dp[i][j]=\max{dp[k][y]+bp[i]×(y-j)} (y-j>=0)\)

首先看第一项,我们观察第二维,j是固定的,那么我们希望ap[i]×y越大越好,于是我们开一个单调队列维护第二维的y,将dp[k][y]+ap[i]×y小的直接丢弃,第二项同理。
于是复杂度就变为了\(O(n×m)\)

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
#define N 2010
using namespace std;

int dp[N][N],q[N],ap[N],bp[N],as[N],bs[N];

int gi() {
  int x=0,o=1; char ch=getchar();
  while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
  if(ch=='-') o=-1,ch=getchar();
  while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
  return o*x;
}

int main() {
  int t=gi(),p=gi(),w=gi(),k,hd,ta;
  for(int i=1; i<=t; i++) ap[i]=gi(),bp[i]=gi(),as[i]=gi(),bs[i]=gi();
  memset(dp,-127,sizeof(dp));
  dp[0][0]=0;  
  for(int i=1; i<=t; i++) {
    for(int j=0; j<=p; j++)
      dp[i][j]=max(dp[i][j],dp[i-1][j]);
    k=max(0,i-w-1);//Mis1:第0天可以对任意天转移 
    q[hd=ta=1]=0;
    for(int j=1; j<=p; j++) {
      while(hd<=ta && j-q[hd]>as[i]) hd++;
      dp[i][j]=max(dp[i][j],dp[k][q[hd]]+ap[i]*(q[hd]-j));
      while(hd<=ta && dp[k][q[ta]]+ap[i]*q[ta]<=dp[k][j]+ap[i]*j) ta--;
      q[++ta]=j;
    }
    q[hd=ta=1]=p;
    for(int j=p-1; j>=0; j--) {
      while(hd<=ta && q[hd]-j>bs[i]) hd++;
      dp[i][j]=max(dp[i][j],dp[k][q[hd]]+bp[i]*(q[hd]-j));
      while(hd<=ta && dp[k][q[ta]]+bp[i]*q[ta]<=dp[k][j]+bp[i]*j) ta--;
      q[++ta]=j;
    }
  }
  printf("%d", dp[t][0]);
  return 0;
}
posted @ 2017-11-10 09:29  HLX_Y  阅读(207)  评论(0编辑  收藏  举报