UVA - 10755——Garbage Heap(三维前缀和)

题目链接：https://vjudge.net/problem/UVA-10755

见白书54页

#include<algorithm>
#include<string.h>
#include<stdio.h>
#define ll long long
using namespace std;
ll t,a,b,c;
ll A[25][25][25];
void split(ll z,ll &b1,ll &b2,ll &b3)
{
    b1=z&1,z>>=1;
    b2=z&1,z>>=1;
    b3=z&1;
}
ll sum(ll x1,ll x2,ll y1,ll y2,ll z1,ll z2)
{
    ll dx=x2-x1+1,dy=y2-y1+1,dz=z2-z1+1,s=0;
    for(ll i=0; i<=7; i++)
    {
        ll b1,b2,b3;
        split(i,b1,b2,b3);
        s+=A[x2-b1*dx][y2-b2*dy][z2-b3*dz]*((b1+b2+b3)&1==1?-1:1);
    }
    return s;
}
int main()
{
    scanf("%lld",&t);
    while(t--)
    {
        ll ans=-100000000000;
        scanf("%lld%lld%lld",&a,&b,&c);
        for(ll i=1; i<=a; i++)
            for(ll j=1; j<=b; j++)
                for(ll k=1; k<=c; k++)
                    scanf("%lld",&A[i][j][k]);
        for(ll i=1; i<=a; i++)
        {
            for(ll j=1; j<=b; j++)
            {
                for(ll k=1; k<=c; k++)
                {
                    for(ll z=1; z<=7; z++)
                    {
                        ll b1,b2,b3;
                        split(z,b1,b2,b3);
                        A[i][j][k]+=A[i-b1][j-b2][k-b3]*((b1+b2+b3)&1==1?1:-1);
                    }
                }
            }
        }
        for(ll x1=1; x1<=a; x1++)
        {
            for(ll x2=x1; x2<=a; x2++)
            {
                for(ll y1=1; y1<=b; y1++)
                {
                    for(ll y2=y1; y2<=b; y2++)
                    {
                        ll M=0;
                        for(ll z=1; z<=c; z++)
                        {
                            ll s=sum(x1,x2,y1,y2,1,z);
                            ans=max(ans,s-M);
                            M=min(M,s);
                        }
                    }
                }
            }
        }
        printf("%lld\n",ans);
        if(t) printf("\n");
    }
}