UVA10755 Garbage Heap --- 降维+dp（最大连续区间和拓展题）

题意： 给定一个axbxc的长方体（a,b,c∈[1,20]），长方体每个格子有个10^9以内的数值，要求找一个子长方体，其格子和最大

题解：Leetcode有道最大连续和的题和这个差不多，只不过Leetcode那个题是一维的，给定一维数组，找一个连续区间，其和最大。 做法是直接dp，dp[i] = max(a[i],dp[i-1]+a[i])。

这道题推广到三维了，根据蓝书上的讲解，这种题可以降维来思考，先不考虑高度，只当作长方形，在长方形上找到一个起点，然后枚举长和宽，这样在o(n^4)的时间复杂度内枚举了所有长方形，然后再考虑高度，在这个长方形的基础上，高度方向上，每一层之和看作一个单独的value，找一个连续的高度区间，使得其和最大即可，这样就把问题转换成了最大连续和，然后o(n)内dp一下就Ok了。 

至于计算每一层某个长方形的和，可以预处理，用sum[i][j][k]，表示第k层中所有1<=x<=i, 1<=y<=j的格子之和，这样在o(n^3)时间复杂度内可以预处理出来。 

 

#include <cstdio>
#include <cstring>
#include <algorithm>
#define MAXN 30
#define INF 1e18
using namespace std;
typedef long long ll;

ll A[MAXN][MAXN][MAXN]; // store cuboid values
ll Sum[MAXN][MAXN][MAXN]; // single layer prefix array
ll dp[MAXN]; // dynamic programming array
int a,b,c; // 3 dimenssion

ll solve() {
    ll res = -INF;
    // find all rectangles 
    // by enumerating start 
    // points as well as their length and width
    for(int x1 = 1;x1 <= a;x1++)
        for(int y1 = 1;y1 <= b;y1++)
            for(int length = 1; length+y1-1 <= b;length++)
                for(int width = 1;width+x1-1 <= a;width++) {
                    int x2 = width+x1-1;
                    int y2 = length+y1-1;
                    memset(dp,0,sizeof(dp));
                    for(int i = 1;i <= c;i++) {
                        ll t = Sum[x2][y2][i]-Sum[x2][y1-1][i]
                               - Sum[x1-1][y2][i]+Sum[x1-1][y1-1][i];
                        dp[i] = max(t,dp[i-1]+t);
                        res = max(res,dp[i]);
                    }

                }

    return res;
}

int main() {
    int T;
    scanf("%d",&T);
    while(T--) {
        memset(dp,0,sizeof(dp));
        memset(Sum,0,sizeof(Sum));
        memset(A,0,sizeof(0));
        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",&A[i][j][k]);
                    Sum[i][j][k] = Sum[i-1][j][k]+Sum[i][j-1][k]-Sum[i-1][j-1][k]+
                                   + A[i][j][k];
                }
        printf("%lld\n",solve());
    }

    return 0;
}