Uva 10755 Garbage Heap（三维最大子矩阵）

题目地址：https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=1696

思路：

二维前缀和，以（i，j）为右下角的矩形中元素的和：sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j]

二维差分：以（x1，y1）（x2，y2）为边界的矩形中元素的和：sum[x2][y2]-sum[x1-1][y2]-sum[x2][y1-1]+sum[x1-1][y1-1]

将长方体按z轴划分，对于平面xoy，枚举边界x1，x2，y1，y2，求得该矩形内元素总和。则按照z轴分层得到a[1],a[2],....,a[C]，子长方体即为按z轴分层所得矩形和a[1],a[2],...,a[C]的子段和，求最大值即可。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define debug
using namespace std;
typedef long long LL;
const int maxn=20+5;
int A,B,C;
LL a[maxn],d[maxn];
LL sum[maxn][maxn][maxn];
LL cube[maxn][maxn][maxn];
int main()
{
#ifdef debu
    freopen("in.txt","r",stdin);
#endif // debug
    int t,cas=0;
    scanf("%d",&t);
    while(t--)
    {
        memset(sum,0,sizeof(sum));
        scanf("%d%d%d",&A,&B,&C);
        for(int i=1; i<=A; i++)
        {
            for(int j=1; j<=B; j++)
            {
                for(int k=1; k<=C; k++)
                {
                    scanf("%lld",&cube[i][j][k]);
                }
            }
        }
        for(int k=1; k<=C; k++)
        {
            for(int i=1; i<=A; i++)
            {
                for(int j=1; j<=B; j++)
                {
                    sum[i][j][k]=sum[i-1][j][k]+sum[i][j-1][k]-sum[i-1][j-1][k]+cube[i][j][k];
                }
            }
        }
        LL ans=-1e12;
        for(int x1=1; x1<=A; x1++)
        {
            for(int x2=x1; x2<=A; x2++)
            {
                for(int y1=1; y1<=B; y1++)
                {
                    for(int y2=y1; y2<=B; y2++)
                    {
                        for(int i=1; i<=C; i++)
                        {
                            a[i]=sum[x2][y2][i]-sum[x1-1][y2][i]-sum[x2][y1-1][i]+sum[x1-1][y1-1][i];
                        }
                        LL tmp=a[1];
                        ans=max(ans,tmp);
                        for(int i=2; i<=C; i++)
                        {
                            if(tmp<=0) tmp=a[i];
                            else tmp+=a[i];
                            ans=max(ans,tmp);
                        }
                    }
                }
            }
        }
        cas++;
        if(cas!=1) printf("\n");
        printf("%lld\n",ans);

    }
    return 0;
}