十分普通的DP+不平凡的转移
这道题状态十分明显。转移是\(O(n^4)\)的,过不去,我们需要优化。
一个十分显然的DP是\(f(i,j)\)表示第\(i\)天时候拥有\(j\)单位股票的最大收益。(可以小于零)。它的转移方式是:
\(f(i,j)=max(f(k,b)+(b-j) \times sell[t] or(buy[t]))\)
此时,我们发现一个转移需要两重循环,也就是\(O(n^2)\),加上右边的式子,就是\(O(n^4)\)的。
考虑优化:
- 对于转移方程,发现我们可以进行参变量分离。对状态转移方程进行代数变形。
-
- \(RHS=max(f(k,b)+b \times sell[t]-j \times sell[t])\)
-
- \(j\)在转移时,是最开始那个等式的\(LHS\),可以提前确定,从\(max\)拿出来。
-
- \(RHS=max(f(k,b)+b \times sell[t])-j\times sell[t]\)
-
- 于是,对于\(max(f(k,b)+b \times sell[t])\),可以拿数据结构维护了。
对于转移方法,我们只需要从\(f(i-W-1,j) , j\le maxP\)转移过来就好了。这一定是最优的那个状态,比它要早的状态一定会$\le f(i-W-1,j) $。
-
- 因为这个(晚的)状态在转移的时候,考虑过 从比他 (早的) 状态直接转移过来,于是\(f(i-W-1,j)\)不小于前面那个。
于是,我们可以将\(f(i-W-1,b) ,b\le maxP\)的所有状态压入某个数据结构。我们发现,对于这种数据结构的要求是:
- 维护最大值
- 可以清理过期状态(买入卖出股票有多少限制)
- 重置方便
就是单调队列吃饭的本领。
一句话解释单调队列:一个状态比你小(限制条件少)还比你强(数值大),那么你就被淘汰了。
单调队列是\(O(n)\)的,枚举\(LHS\)的\(i\)是\(O(n)\)的。总复杂度\(O(n^2)\),可以过了。
#include#include #include #include #include #include #include #include