题意:
求三维最大矩阵和。
思路:
先递推求出每个二维矩阵的和,然后枚举第3维z,在第3维上进行最大子段和dp[i]=max(dp[i-1]+c[i],c[i]),维护答案ans=max(ans,dp[i])。
#include<cstdio>
#include<cstring>
#include<algorithm>
#define FOR(i,s,t) for(int i=(s);i<=(t);i++)
using namespace std;
typedef long long LL;
const LL INF=1LL<<60;
const int MAX=25;
LL S[MAX][MAX][MAX],A[MAX][MAX][MAX],C[MAX],dp[MAX];
int main(){
int T;
scanf("%d",&T);
while(T--){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
memset(S,0,sizeof(S));
FOR(x,1,a) FOR(y,1,b) FOR(z,1,c) scanf("%lld",&A[x][y][z]);
FOR(x,1,a) FOR(y,1,b) FOR(z,1,c){
S[x][y][z]=S[x][y-1][z]+S[x-1][y][z]-S[x-1][y-1][z]+A[x][y][z];
}
LL ans=-INF;
FOR(x1,1,a) FOR(x2,x1,a) FOR(y1,1,b) FOR(y2,y1,b){
memset(C,0,sizeof(C));
FOR(z,1,c){
C[z]=S[x2][y2][z]-S[x1-1][y2][z]-S[x2][y1-1][z]+S[x1-1][y1-1][z];
}
dp[0]=0;
FOR(z,1,c){
dp[z]=max(dp[z-1]+C[z],C[z]);
ans=max(ans,dp[z]);
}
}
printf("%lld\n",ans);
if(T) printf("\n");
}
return 0;
}