「学习笔记」单调队列优化dp

2023-05-02,,

目录
算法
例题
最大子段和
题意
思路
代码
修剪草坪
题意
思路
代码
瑰丽华尔兹
题意
思路
代码
股票交易
题意
思路
代码

算法

使用单调队列优化dp 废话

对与一些dp的转移方程,我们可以通过拆使它与某个区间的最值相关。

这时可以用单调队列算出区间最值,进行优化。

例题

最大子段和

题意

给出一个长度为 \(n\) 的整数序列,从中找出一段长度不超过 \(m\) 的连续子序列,使得整个序列的和最大。

思路

设 \(sum_i\) 为 \(i\) 的前缀和,易得答案为:

\[\max_\limits{1\le i\le n}\{sum_i-\min_\limits{i-m\le k\le i-1}\{sum_k\}\}
\]

其中 \(\min_\limits{i-m\le k\le i-1}\{sum_k\}\) 这部分可以用单调队列快速求出。

那么算起来就变得简单多了。

代码

点击查看代码
#include<bits/stdc++.h>
#define _for(i,a,b) for(int i=a;i<=b;++i)
#define for_(i,a,b) for(int i=a;i>=b;--i)
#define ll long long
using namespace std;
const int N=3e5+10;
int n,m,a[N],sum[N],f[N],ans;
int q[N],h=1,t=0;
inline int rnt(){
int x=0,w=1;char c=getchar();
while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
while(c>='0'&&c<='9')x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x*w;
}
void tmp(int k){
while(h<=t&&q[h]<k-m)++h;
ans=max(ans,sum[k]-sum[q[h]]);
while(h<=t&&sum[q[t]]>sum[k])--t;
q[++t]=k;
}
int main(){
n=rnt(),m=rnt();
_for(i,1,n){
a[i]=rnt();
sum[i]=sum[i-1]+a[i];
tmp(i);
}
printf("%d\n",ans);
return 0;
}

修剪草坪

题意

给定一个 \(n\ (1\le n\le 10^5)\) 个元素的序列,可任选多个数,要求任意一段连续选的数长度不能超过 \(k\)。

\(1\le N\le 10^5\)

思路

设 \(f_{i,0/1}\) 表示第 \(i\) 个数选(1)或是不选(0),\(sum_i\) 表示 \(i\) 的前缀和。

转移方程为:

\[f_{i,0}=\max\{f_{i-1,0},f_{i-1,1}\}\\\
f_{i,1}=\max_\limits{i-k\le j<i}\{f_{j,0}+(sum_i-sum_j)\}
\]

\(f_{i,1}\) 转移方程可以拆成:

\[f_{i,1}=sum_i+\max_\limits{i-k\le j<i}\{f_{j,0}-sum_j\}
\]

用单调队列维护 \(f_{j,0}-sum_j\) 即可。

代码

点击查看代码
#include<bits/stdc++.h>
#define _for(i,a,b) for(ll i=a;i<=b;++i)
#define for_(i,a,b) for(ll i=a;i>=b;--i)
#define ll long long
using namespace std;
const ll N=3e5+10;
ll n,k,a[N],sum[N],f[N][2],ans;
ll q[N],h=1,t=0;
inline ll rnt(){
ll x=0,w=1;char c=getchar();
while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
while(c>='0'&&c<='9')x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x*w;
}
void tmp(ll i){
while(h<=t&&q[h]<i-k)++h;
f[i][1]=f[q[h]][0]+sum[i]-sum[q[h]];
while(h<=t&&f[q[t]][0]-sum[q[t]]<f[i][0]-sum[i])--t;
q[++t]=i;
}
int main(){
n=rnt(),k=rnt();
tmp(0);
_for(i,1,n){
a[i]=rnt();
sum[i]=sum[i-1]+a[i];
f[i][0]=max(f[i-1][0],f[i-1][1]);
tmp(i);
}
printf("%lld\n",max(f[n][0],f[n][1]));
return 0;
}

瑰丽华尔兹

题意

给出一个 \(N*M\) 的船上地图,有空地也有家具。

船上有 \(K\) 段时间,在每段时间都会往不同的方向倾斜,钢琴也会朝着那个方向倾斜,但不允许碰上家具。

求钢琴最长的滑行距离。

\(1\le N,M\le200,K\le200\)

思路

设 \(f_{i,j,k}\) 表示在第 \(i\) 段时间滑行到 \(j,k\) 的位置。

那么转移方程就是:

\[f_{i,j,k}=\max_\limits{(jj,kk)是i时刻可以滑到(j,k)的点}{f_{i,jj,kk}+|jj-j|+|kk-k|}
\]

瞎写的

按着倾斜的方向遍历,再用单调队列优化优化就好了。

代码

点击查看代码
#include<bits/stdc++.h>
#define _for(i,a,b) for(int i=a;i<=b;++i)
#define for_(i,a,b) for(int i=a;i>=b;--i)
#define ll long long
using namespace std;
const int N=210,inf=0x3f3f3f3f;
int n,m,x,y,k,mp[N][N];
int f[N][N][N],la[N][N][N],ans;
inline ll rnt(){
ll x=0,w=1;char c=getchar();
while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
while(c>='0'&&c<='9')x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x*w;
}
inline char rch(){
char c=getchar();
while(c!='.'&&c!='x')c=getchar();
return c;
}
struct dq{//deque
int q[N],h,t;
void nw(){
memset(q,0,sizeof(q));
h=1,t=0;
}
bool empty(){return h>t;}
int front(){return q[h];}
int back(){return q[t];}
void pop_f(){++h;}
void pop_b(){--t;}
void push(int x){q[++t]=x;}
};
void dp(int d,int s,int t,int fx){
int len=(t-s+1);
if(fx==1){
_for(j,1,m){
dq q;q.nw();
for_(i,n,1){
if(mp[i][j]){
while(!q.empty())q.pop_f();
continue;
}
while(!q.empty()&&q.front()>i+len)q.pop_f();
while(!q.empty()&&f[d-1][q.back()][j]+q.back()-i<f[d-1][i][j])q.pop_b();
q.push(i);
if(f[d-1][q.front()][j]>-1)
f[d][i][j]=f[d-1][q.front()][j]+q.front()-i;
ans=max(ans,f[d][i][j]);
}
}
}
else if(fx==2){
_for(j,1,m){
dq q;q.nw();
_for(i,1,n){
if(mp[i][j]){
while(!q.empty())q.pop_f();
continue;
}
while(!q.empty()&&q.front()<i-len)q.pop_f();
while(!q.empty()&&f[d-1][q.back()][j]+i-q.back()<f[d-1][i][j])q.pop_b();
q.push(i);
if(f[d-1][q.front()][j]>-1)
f[d][i][j]=f[d-1][q.front()][j]+i-q.front();
ans=max(ans,f[d][i][j]);
}
}
}
else if(fx==3){
_for(i,1,n){
dq q;q.nw();
for_(j,m,1){
if(mp[i][j]){
while(!q.empty())q.pop_f();
continue;
}
while(!q.empty()&&q.front()>j+len)q.pop_f();
while(!q.empty()&&f[d-1][i][q.back()]+q.back()-j<f[d-1][i][j])q.pop_b();
q.push(j);
if(f[d-1][i][q.front()]>-1)
f[d][i][j]=f[d-1][i][q.front()]+q.front()-j;
ans=max(ans,f[d][i][j]);
}
}
}
else{
_for(i,1,n){
dq q;q.nw();
_for(j,1,m){
if(mp[i][j]){
while(!q.empty())q.pop_f();
continue;
}
while(!q.empty()&&q.front()<j-len)q.pop_f();
while(!q.empty()&&f[d-1][i][q.back()]+j-q.back()<f[d-1][i][j])q.pop_b();
q.push(j);
if(f[d-1][i][q.front()]>-1)
f[d][i][j]=f[d-1][i][q.front()]+j-q.front();
ans=max(ans,f[d][i][j]);
}
}
}
}
int main(){
n=rnt(),m=rnt(),x=rnt(),y=rnt(),k=rnt();
_for(i,1,n)
_for(j,1,m)
mp[i][j]=(bool)(rch()=='x');
memset(f,-inf,sizeof(f));
f[0][x][y]=0;
_for(i,1,k){
int s=rnt(),t=rnt(),fx=rnt();
dp(i,s,t,fx);
}
printf("%d\n",ans);
return 0;
}

股票交易

题意

一共有 \(T\) 天,知道了每天股票的买入金额 \(ap_i\),卖出金额 \(bp_i\),买入限制 \(as_j\),卖出限制 \(bs_j\)。要求持有股票数不能超过 \(MaxP\) ,两次交易相隔 \(w\) 天。

\(1\le w<T\le 2*10^3,1\le MaxP\le 2*10^3\)

思路

设 \(f_{i,j}\) 表示第 \(i\) 天持有 \(j\) 个股票的最大钱数。

有四种情况:

凭空买

转移方程:
\[f_{i,j}=-ap_i*j
\]
不买不卖

转移方程:
\[f_{i,j}=\max\{f_{i,j},f_{i-1,j}\}
\]
只买

转移方程:
\[f_{i,j}=\max\{f_{i,j},\max_\limits{j-as_i\le k\le j}\{f_{i-w-1,k}-(j-k)*ap_i\}\}
\]
只卖

转移方程:
\[f_{i,j}=\max\{f_{i,j},\max_\limits{j\le k\le j+bs_i}\{f_{i-w-1,k}+(k-j)*bp_i\}\}
\]

此时时间复杂度是 \(O(T*MaxP^2)\)。

观察只买和只卖的情况,可以发现它们的转移方程可以用单调队列优化掉一维。

那么时间复杂度就被优化成了 \(O(T*MaxP)\),完全可过。

代码

点击查看代码
#include<bits/stdc++.h>
#define _for(i,a,b) for(ll i=a;i<=b;++i)
#define for_(i,a,b) for(ll i=a;i>=b;--i)
#define ll long long
using namespace std;
const ll N=4010,inf=0x3f3f3f3f;
ll t,mxp,w,ap[N],bp[N],as[N],bs[N];
inline ll rnt(){
ll x=0,w=1;char c=getchar();
while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
while(c>='0'&&c<='9')x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x*w;
}
struct dq{//deque
ll q[N],h,t;
void nw(){
memset(q,0,sizeof(q));
h=1,t=0;
}
bool empty(){return h>t;}
ll front(){return q[h];}
ll back(){return q[t];}
void pop_f(){++h;}
void pop_b(){--t;}
void push(int x){q[++t]=x;}
};
namespace SOLVE{
ll f[N][N];
void DpPkm(ll i){//凭空买
_for(j,0,min(mxp,as[i]))
f[i][j]=-ap[i]*j;
return;
}
void DpBmbm(ll i){//不买也不卖
_for(j,0,mxp)
f[i][j]=max(f[i][j],f[i-1][j]);
return;
}
void DpBuy(ll i){//原基础上买来
dq q;q.nw();
_for(j,0,mxp){
while(!q.empty()&&j-q.front()>as[i])q.pop_f();
while(!q.empty()&&f[i-w-1][j]>f[i-w-1][q.back()]-(j-q.back())*ap[i])q.pop_b();
q.push(j);
f[i][j]=max(f[i][j],f[i-w-1][q.front()]-(j-q.front())*ap[i]);
}
return;
}
void DpSell(ll i){//原基础上卖出
dq q;q.nw();
for_(j,mxp,0){
while(!q.empty()&&q.front()-j>bs[i])q.pop_f();
while(!q.empty()&&f[i-w-1][j]>f[i-w-1][q.back()]+(q.back()-j)*bp[i])q.pop_b();
q.push(j);
f[i][j]=max(f[i][j],f[i-w-1][q.front()]+(q.front()-j)*bp[i]);
}
return;
}
ll Solve(){
memset(f,-inf,sizeof(f));
_for(i,1,t){
DpPkm(i);
DpBmbm(i);
if(i-w>1){
DpBuy(i);
DpSell(i);
}
}
return f[t][0];
}
}
int main(){
t=rnt(),mxp=rnt(),w=rnt();
_for(i,1,t)
ap[i]=rnt(),bp[i]=rnt(),as[i]=rnt(),bs[i]=rnt();
printf("%lld\n",SOLVE::Solve());
return 0;
}/* */

$$
\Huge{\mathfrak{The\ End}}
$$

学习笔记」单调队列优化dp的相关教程结束。

《「学习笔记」单调队列优化dp.doc》

下载本文的Word格式文档,以方便收藏与打印。