bzoj1177 [Apio2009]Oil(枚举)

求三个不相交的k*k的正方形能覆盖的最大权值和。
各种优化枚举即可。详见portal
注意比较坑的是bzoj数据大概有些问题,读入优化会挂掉qaq

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 1510
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,k,a[N][N],s[N][N],sum[N][N],f1[N][N],f2[N][N],f3[N][N],f4[N][N];//f1--(1,1)-(i,j) f2--(i,j)-(1,m)  f3--(n,1)-(i,j)  f4--(i,j)-(n,m)
int mx1[N],mx2[N];
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();k=read();int ans=0;
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j) a[i][j]=read();
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j) sum[i][j]=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1]+a[i][j];
    for(int i=k;i<=n;++i)
        for(int j=k;j<=m;++j) s[i][j]=sum[i][j]-sum[i-k][j]-sum[i][j-k]+sum[i-k][j-k];
    for(int i=k;i<=n;++i)
        for(int j=k;j<=m;++j) f1[i][j]=max(s[i][j],max(f1[i-1][j],f1[i][j-1]));
    for(int i=k;i<=n;++i)
        for(int j=m-k+1;j>=1;--j) f2[i][j]=max(s[i][j+k-1],max(f2[i-1][j],f2[i][j+1]));
    for(int i=n-k+1;i>=1;--i)
        for(int j=k;j<=m;++j) f3[i][j]=max(s[i+k-1][j],max(f3[i+1][j],f3[i][j-1]));
    for(int i=n-k+1;i>=1;--i)
        for(int j=m-k+1;j>=1;--j) f4[i][j]=max(s[i+k-1][j+k-1],max(f4[i+1][j],f4[i][j+1]));
    for(int i=k;i<=n-k;++i)
        for(int j=k;j<=m-k;++j){
            ans=max(ans,f1[i][j]+f2[i][j+1]+f3[i+1][m]);
            ans=max(ans,f1[i][m]+f3[i+1][j]+f4[i+1][j+1]);
            ans=max(ans,f1[n][j]+f2[i][j+1]+f4[i+1][j+1]);
            ans=max(ans,f1[i][j]+f3[i+1][j]+f4[1][j+1]);
        }
    for(int i=k;i<=n;++i)
        for(int j=k;j<=m;++j) mx1[i]=max(mx1[i],s[i][j]),mx2[j]=max(mx2[j],s[i][j]);
    for(int i=k;i<=n-k-k;++i)
        for(int j=i+k,mid=mx1[j];j<=n-k;++j,mid=max(mid,mx1[j])) ans=max(ans,mid+f1[i][m]+f3[j+1][m]);
    for(int i=k;i<=m-k-k;++i)
        for(int j=i+k,mid=mx2[j];j<=m-k;++j,mid=max(mid,mx2[j])) ans=max(ans,mid+f1[n][i]+f2[n][j+1]);
    printf("%d\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/icefox_zhx/article/details/80431233