cf 1304F2 (单调队列优化DP)

cf 1304F2 (单调队列优化DP)

题意:

给你 n n n天,每天有 m m m个拍照点,每次拍照是连续两天且拍照颜色不同,每次最多拍连续k段,连续两天就是2*k,每个拍照点有一个权值,求最大权值和

思路:

容易想到一个 d p dp dp d p [ i ] [ j ] dp[i][j] dp[i][j] i i i天拍完了,第 i i i天从 j j j这个地方开始拍,注意这个在转移前是一个相机的前一天,但是转移后得把一个相机的另一天计算进去。

d p [ i ] [ j ] = m a x ( d p [ i − 1 ] [ l ] + s u m [ m i n ( j + k − 1 , m ) ] − s u m [ j − 1 ] − 重 合 部 分 ) dp[i][j]=max(dp[i-1][l]+sum[min(j+k-1,m)]-sum[j-1]-重合部分) dp[i][j]=max(dp[i1][l]+sum[min(j+k1,m)]sum[j1])

这样的复杂度是 O ( n m 2 ) O(nm^2) O(nm2)的,显然不能通过

我们观察到把 d p dp dp的第一维去掉后是一个 1 D / 1 D 1D/1D 1D/1D的dp模型,考虑格子间的移动,很明显前面和后面上下格子不相交的话,可以通过维护前后缀 d p dp dp最大值来解决。如果相交的话,

d p [ i ] [ j ] = d p [ i − 1 ] [ l ] + s u m [ i ] [ m i n ( j + k − 1 , m ) ] − s u m [ i ] [ l − 1 ] dp[i][j]=dp[i-1][l]+sum[i][min(j+k-1,m)]-sum[i][l-1] dp[i][j]=dp[i1][l]+sum[i][min(j+k1,m)]sum[i][l1]

显然可以用拆成 l , j l,j l,j分开的形式,每个决策入队出队一次,就可以做到 O ( n m ) O(nm) O(nm)了,用单调队列优化即可,正反都做一遍,注意这里 j j j是可以取到的,所以有一个得丢进去再更新,这题就没了,注意这里 d p dp dp值更新前后都是只算每个相机的第一天,前后缀最大值才保留2天的

#include<bits/stdc++.h>
#define fi first 
#define se second
using namespace std;
typedef pair<int,int>P;
const int maxn=2e4+10;
int dp[53][maxn],mxp[53][maxn],mxs[53][maxn],n,m,a[53][maxn],k,sum[53][maxn],h1,h2,t1,t2;
P q1[maxn],q2[maxn];
int main(){
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j)scanf("%d",&a[i][j]),sum[i][j]=sum[i][j-1]+a[i][j];
    for(int i=1;i<=m;++i)dp[1][i]=sum[1][min(i+k-1,m)]-sum[1][i-1];
    for(int i=1;i<=m;++i)mxp[1][i]=max(mxp[1][i-1],dp[1][i]+sum[2][min(i+k-1,m)]-sum[2][i-1]);
    for(int i=m;i>=1;--i)mxs[1][i]=max(mxs[1][i+1],dp[1][i]+sum[2][min(i+k-1,m)]-sum[2][i-1]);
    for(int i=2;i<=n;++i){
        for(int j=1;j<=m;++j){
            if(j+k<=m)
                dp[i][j]=max(dp[i][j],mxs[i-1][j+k]+sum[i][min(j+k-1,m)]-sum[i][j-1]);
            if(j-k>=1)
                dp[i][j]=max(dp[i][j],mxp[i-1][j-k]+sum[i][min(j+k-1,m)]-sum[i][j-1]);
        }
        h1=h2=1;t1=t2=0;
        for(int j=1;j<=m;++j){
            while(h1<=t1&&q1[h1].fi<j-k+1)h1++;
            int num=dp[i-1][j]-sum[i][j-1];
            while(h1<=t1&&q1[t1].se<=num)t1--;
            q1[++t1]={j,num};
            if(h1<=t1)dp[i][j]=max(dp[i][j],q1[h1].se+sum[i][min(m,j+k-1)]);
        }
        for(int j=m;j>=1;--j){
            while(h2<=t2&&q2[h2].fi>j+k-1)h2++;
            if(h2<=t2)dp[i][j]=max(dp[i][j],q2[h2].se-sum[i][j-1]);
            int num=dp[i-1][j]+sum[i][min(j+k-1,m)];
            while(h2<=t2&&q2[t2].se<=num)t2--;
            q2[++t2]={j,num};
        }
        for(int j=1;j<=m;++j){
            mxp[i][j]=max(mxp[i][j-1],dp[i][j]+sum[i+1][min(m,j+k-1)]-sum[i+1][j-1]);
        }
        for(int j=m;j>=1;--j){
            mxs[i][j]=max(mxs[i][j+1],dp[i][j]+sum[i+1][min(m,j+k-1)]-sum[i+1][j-1]);
        }
    }
    int ans=0;
    for(int i=1;i<=m;++i){
        if(i+k-1>m)break;
        ans=max(ans,dp[n][i]);
    }
    cout<<ans<<"\n";
    return 0;
}

启示:

1D/1D d p dp dp模型形如当前转移点是下面上方不相交的这种,取min/max的时候,常可以考虑维护前后缀最值

下方有相交的,若能拆开两个变量,则考虑单调队列优化在这里插入图片描述

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值