【笔记篇】单调队列优化dp学习笔记&&luogu2569/bzoj1855股票交♂易

DP颂

DP之神 圣洁美丽 算法光芒照大地

我们怀着 崇高敬意 跪倒在DP神殿里

你的复杂 能让蒟蒻 试图入门却放弃

在你光辉 照耀下面 AC真心不容易

dp大概是最经久不衰 亘古不化的算法了吧.
而且有各种各样的类型 优化之类的.
一直dp都不怎么好. 而且也不太知道应该怎么提高.
基本见到不认识的dp方程就不大会推(但我会打表啊= =

所以dp还是很有的学的~

正好最近刚刚肝了计算几何, 所以就顺带搞一下斜率优化dp一类的…

单调队列优化dp

单调队列大家都会吧?
不会的先出去学一下, 这里不讲.

=====================================================================

好的, 我们来看一下这个柿子

$$f[i]=max\{f[j]+\omega(j)\} (j\in[1..i))$$

其中$\omega(j)$是一个费用函数, 一般会根据题目的不同而变化.

这个dp能做到什么复杂度呢?
首先一眼$O(n^2)$…
然而我们可以用一个变量记录一下之前出现过的最大值.
这样转移是$O(1)$的了, 总复杂度就降到了$O(n)$.

但是如果是这样呢?

$$f[i]=max\{f[j]+\omega(j)\}(j\in[i-m,i))$$

那就不能只维护一个变量了, 因为最大值如果出现在$j$的取值区间之外则转移是不合法的.
这样我们就考虑用单调队列来维护最大值, 这样转移依然可以做到$O(1)$, 总复杂度$O(n)$.

看道题: (woc辣鸡bzoj给的什么zz数据范围, T都没给怎么做…)

这题可以写出这么一个状态转移方程

$$令f[i][j]表示第i天拥有j支股票的最大收益, \\ f[i][j]=max\left\{ \begin{matrix} f[i-w-1][k]-ap[i]*(j-k), (k\in[j-as[i],j)) //买入\\ f[i-w-1][k]+bp[i]*(k-j),(k\in(j,j+bs[i]]) //卖出\\ f[i-1][j]//不交♂易 \end{matrix} \right.$$

其中不交易的情况好处理, 但是如果前面两种枚举$k$的话就要做到$O(n^*maxP^2)$, 显然是过不了的, 我们必须考虑优化.
我们以买入为栗化一波柿子(因为卖出同理) :

$$f[i][j]=max\{f[i-w-1][k]-ap[i]*(j-k)\}\\ =max\{f[i-w-1][k]+ap[i]*k\}-ap[i]*j (k\in[j-as[i],j])$$
我们令 $\omega(x)=ap[i]*x$, 而我们枚举 $i$, 就可以视为$i$是定值, 于是 $ap[i],as[i]$都是定值.
我们就可以看出第二维形成了一个能用单调队列优化的柿子了.
这样优化之后复杂度成功降到了 $O(n*maxP)$, 就可以通过此题了.

根据贪心原则, 为了获得最多的现金, 手里不应该留股票, 所以用每个$f[i][0]$更新答案即可.

不过要注意一下边界条件… 挺扯淡的..

代码:

#include <cstdio>
#include <cstring>
const int N=2020;
const int INF=0x7fffffff;
int q[N<<1],d[N<<1],h=1,t=0;
int f[N][N],ap[N],bp[N],as[N],bs[N];
inline int gn(int a=0,char c=0){
    for(;c<'0'||c>'9';c=getchar());
    for(;c>47&&c<58;c=getchar())a=a*10+c-48;return a;
}
inline int max(const int& a,const int& b){
    return a>b?a:b;
}
int main(){
    int n=gn(),m=gn(),w=gn(),ans=0;
    for(int i=1;i<=n;++i)
        ap[i]=gn(),bp[i]=gn(),as[i]=gn(),bs[i]=gn();
    memset(f,192,sizeof(f));
    for(int i=1;i<=n;++i){
        for(int j=0;j<=as[i];++j) f[i][j]=-ap[i]*j;
        for(int j=0;j<=m;++j) f[i][j]=max(f[i][j],f[i-1][j]);
        if(i>w){
                h=1; t=0;
            for(int j=0;j<=m;++j){              
                int val=f[i-w-1][j]+j*ap[i];
                while(t>=h&&val>=q[t]) --t;
                q[++t]=val; d[t]=j;
                while(t>=h&&d[h]<j-as[i]) ++h;
                f[i][j]=max(f[i][j],q[h]-ap[i]*j);
            }
            h=1; t=0;
            for(int j=m;j>=0;--j){
                int val=f[i-w-1][j]+bp[i]*j;
                while(t>=h&&val>=q[t]) --t;
                q[++t]=val; d[t]=j;
                while(t>=h&&d[h]>j+bs[i]) ++h;
                f[i][j]=max(f[i][j],q[h]-bp[i]*j);
            }
        }
        ans=max(ans,f[i][0]);
    }
    printf("%d",ans);
}

反正差不多就这样吧….