HDU1892 二维树状数组区间求和,单点修改

二维树状数组详解

最新推荐文章于 2022-01-13 08:42:06 发布

原创最新推荐文章于 2022-01-13 08:42:06 发布 · 262 阅读

0 ·

CC 4.0 BY-SA版权

树状数组专栏收录该内容

4 篇文章

订阅专栏

本文详细介绍了二维树状数组的应用，包括如何实现区间查询和单点更新操作，并提供了一个完整的C++代码示例，用于解决二维树状数组的模板问题。

传送门：二维树状数组模板题

分析：二维树状数组模板题，求区间和跟单点修改

代码如下：

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

const int maxn = 1000+10;
const int N = 1005;
int a[maxn][maxn];
int c[maxn][maxn];

inline int lowbit(int x) {return x&(-x); }
inline void update(int x, int y, int val) {
    for (int i=x; i<=N; i += lowbit(i))
        for (int j=y; j<=N; j += lowbit(j)) c[i][j] += val;
}

inline int query(int x, int y) {
    int sum = 0;
    for (int i=x; i>0; i -= lowbit(i))
        for (int j=y; j>0; j -= lowbit(j) ) sum += c[i][j];
    return sum;
}

inline int Min(int x, int y) {return x<y?x:y; }

int main(){
    int T;
    scanf("%d",&T);
    int kase = 0;
    while (T--) {
       printf("Case %d:\n",++kase);
       memset(c,0,sizeof(c));
       for (int i=1; i<=N; i++)
         for (int j=1; j<=N; j++) update(i,j,1);
       for (int i=1; i<=N; i++)
            for (int j=1; j<=N; j++) a[i][j] = 1;

       int q;
       int X1,Y1,X2,Y2;
       int x,y,val;
       char s[10];
       scanf("%d",&q);

       while (q--) {
          scanf("%s",s);
          if (s[0] == 'S') {
               scanf("%d%d%d%d",&X1,&Y1,&X2,&Y2);
               if (X1>X2) swap(X1,X2);
               if (Y1>Y2) swap(Y1,Y2);
               X1++; Y1++; X2++; Y2++;
               printf("%d\n",query(X2,Y2)-query(X2,Y1-1)-query(X1-1,Y2)+query(X1-1,Y1-1));
          }
          else if (s[0] == 'A') {
               scanf("%d%d%d",&x,&y,&val);
               x++; y++;
               update(x,y,val);
               a[x][y] += val;
          }
          else if (s[0] == 'D' ){
              scanf("%d%d%d",&x,&y,&val);
              x++; y++;
              val = Min(a[x][y],val);
              update(x,y,-val);
              a[x][y] -= val;
          }
          else {
              scanf("%d%d%d%d%d",&X1,&Y1,&X2,&Y2,&val);
              X1++; Y1++; X2++; Y2++;
              val = Min(a[X1][Y1],val);
              update(X1,Y1,-val);
              update(X2,Y2,val);
              a[X1][Y1] -= val;
              a[X2][Y2] += val;
          }

       }
    }
    return 0;
}