HOJ 2686 Magic-Box

三维树状数组,关键是更新时,空间想象比二维复杂些。其原理都是相同的!

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN =210;

int c[MAXN][MAXN][MAXN];

inline int lowbit(int x)
{
     return x & (-x);
}

void Modify(int x, int y, int z, int data)
{
     for(int i = x; i < MAXN; i += lowbit( i ))
          for(int j = y; j < MAXN; j += lowbit( j ))
               for(int k = z; k < MAXN; k += lowbit( k ))
                    c[i][j][k] += data;
}

int GetSum(int x, int y, int z)
{
     int sum = 0;
     for(int i = x; i >= 1; i -= lowbit( i ))
          for(int j = y; j >= 1; j -= lowbit( j ))
               for(int k = z; k >= 1; k -= lowbit( k ))
                    sum += c[i][j][k];
     return sum;
}
int main()
{
     int t;
     char s[10];
     int x1, y1, z1, x2, y2, z2, k;
     int x, y, z;
     while(scanf("%d", &t) != EOF)
     {
          memset(c, 0, sizeof(c));
          while(t--)
          {
               scanf("%s", s);
               if(s[0] == 'A')
               {
                    scanf("%d %d %d %d %d %d %d", &x1, &y1, &z1, &x2, &y2, &z2, &k);
                    Modify(x2+1, y2+1, z2+1, -k);
                    Modify(x1, y2+1, z2+1, k);
                    Modify(x2+1, y1, z2+1, k);
                    Modify(x2+1, y2+1, z1, k);
                    Modify(x2+1, y1, z1, -k);
                    Modify(x1, y2+1, z1, -k);
                    Modify(x1, y1, z2+1, -k);
                    Modify(x1, y1, z1, k);
               }
               else if(s[0] == 'Q')
               {
                    scanf("%d %d %d", &x, &y, &z);
                    printf("%d\n", GetSum(x, y, z));
               }
          }
     }
    return 0;
}


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值