Codeforces 1304 F1 Animal Observation (easy version) —— DP+线段树

This way

题意：

给你一个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;
}