【UVa】10755 Garbage Heap 三维前缀和

题目分析：将前缀和应用到三维，求最大子矩阵。为S[x][y][z]数组中每个坐标保存从(0,0,0)到(x,y,z)范围内的子矩阵的元素和，最后用多次区间加减法可以得到需要的子矩阵的元素和，再用类似一维求最大连续和的方法求三维最大连续和。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;

#define REP( i , a , b ) for ( int i = a ; i < b ; ++ i )
#define FOR( i , a , b ) for ( int i = a ; i <= b ; ++ i )
#define REV( i , a , b ) for ( int i = a ; i >= b ; -- i )
#define CLR( a , x ) memset ( a , x , sizeof a )
#define SIZE MAXN][MAXN][MAXN
#define FFF FOR ( x , 1 , A ) FOR ( y , 1 , B ) FOR ( z , 1 , C )

typedef long long LL ;

const int MAXN = 21 ;
const LL INF = 1e18 ;

LL S[SIZE] ;
int A , B , C ;

LL calculate ( int x1 , int x2 , int y1 , int y2 , int z1 , int z2 ) {
	LL res = 0 ;
	res += S[x2    ][y2    ][z2    ] ;
	res -= S[x1 - 1][y2    ][z2    ] ;
	res -= S[x2    ][y1 - 1][z2    ] ;
	res -= S[x2    ][y2    ][z1 - 1] ;
	res += S[x1 - 1][y2    ][z1 - 1] ;
	res += S[x2    ][y1 - 1][z1 - 1] ;
	res += S[x1 - 1][y1 - 1][z2    ] ;
	res -= S[x1 - 1][y1 - 1][z1 - 1] ;
	return res ;
}

void solve () {
	scanf ( "%d%d%d" , &A , &B , &C ) ;
	CLR ( S , 0 ) ;
	FFF scanf ( "%lld" , &S[x][y][z] ) ;
	FFF S[x][y][z] += S[x - 1][y][z] ;
	FFF S[x][y][z] += S[x][y - 1][z] ;
	FFF S[x][y][z] += S[x][y][z - 1] ;
	//FFF printf ( "S[%d][%d][%d] = %lld\n" , x , y , z , S[x][y][z] ) ;
	LL ans = -INF ;
	FOR ( x1 , 1 , A )
		FOR ( x2 , x1 , A )
			FOR ( y1 , 1 , B )
				FOR ( y2 , y1 , B ) {
					LL M = 0 ;
					FOR ( z , 1 , C ) {
						LL sum = calculate ( x1 , x2 , y1 , y2 , 1 , z ) ;
						ans = max ( ans , sum - M ) ;
						M = min ( M , sum ) ;
					}
				}
	printf ( "%lld\n" , ans ) ;
}

int main () {
	int T ;
	scanf ( "%d" , &T ) ;
	while ( T -- ) {
		solve () ;
		if ( T )
			putchar ( '\n' ) ;
	}
	return 0 ;
}