题意:
给你一个n*m的矩阵,每一行你都可以放置一个2k大小的矩阵,连着i和i+1行,第n行只能放置一个1k的矩阵。问你最终被所有矩阵包括的值得和最大是多少。
题解:
明显是个DP,因为它之后做的决策对之前无影响并且可以很容易写出dp方程:
dp[i][j]表示到第i行,选择的矩阵的开头是j时,答案最大是多少。(由于之后处理方便我使用了滚动数组)
那么从dp[i-1]转移过来,但是肯定不能一一枚举,否则时间复杂度不够,所以大致可以将上个区间分成三块:
两种不相交的和一种相交的区间,那么不相交的区间只需要用线段树查询最大值即可,对于相交的区间,由于k只有20,所以暴力枚举即可。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e4+5;
ll a[55][N],sum[55][N],mx[N*4],dp[2][N];
void build(int l,int r,int root){
if(l==r){
mx[root]=dp[0][l];
return ;
}
int mid=l+r>>1;
build(l,mid,root<<1);
build(mid+1,r,root<<1|1);
mx[root]=max(mx[root<<1],mx[root<<1|1]);
}
ll query(int l,int r,int root,int ql,int qr){
if(l>=ql&&r<=qr)
return mx[root];
int mid=l+r>>1;
ll ans=0;
if(mid>=ql)
ans=query(l,mid,root<<1,ql,qr);
if(mid<qr)
ans=max(ans,query(mid+1,r,root<<1|1,ql,qr));
return ans;
}
int main()
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
ll ans=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
scanf("%lld",&a[i][j]),sum[i][j]=sum[i][j-1]+a[i][j];
for(int i=1;i<=n;i++){
for(int j=1;j<=m-k+1;j++){
ll v=sum[i][j+k-1]-sum[i][j-1];
if(i!=n)
v+=sum[i+1][j+k-1]-sum[i+1][j-1];
if(i==1){
dp[1][j]=v;
ans=max(ans,v);
continue;
}
if(j>k)
dp[1][j]=query(1,m,1,1,j-k)+v;
if(j+k<=m)
dp[1][j]=max(dp[1][j],query(1,m,1,j+k,m)+v);
int l;
ll s;
for(l=max(1,j-k+1),s=sum[i][l+k-1-1]-sum[i][j-1];l<=j+k-1&&l+k-1<=m;l++){
if(l<=j)
s+=a[i][l+k-1];
else
s-=a[i][l-1];
dp[1][j]=max(dp[1][j],dp[0][l]+v-s);
ans=max(ans,dp[1][j]);
}
}
for(int j=1;j<=m-k+1;j++)
dp[0][j]=dp[1][j],dp[1][j]=0;
build(1,m,1);
}
printf("%lld\n",ans);
return 0;
}