[模板]二维树状树组 单点修改 区间查询

输入格式:

1.输入两个整数n,m,表示矩阵的长和宽;

2.输入n行m列个整数,为矩阵的每个元素;

3.输入一个整数T,表示操作次数;

4.输入T行:

格式1:1 x y k,表示将坐标为(x,y)的点加上k;

格式2:2 x1 y1 x2 y2,表示求左上角坐标为(x1,y1),右下角坐标为(x2,y2)的矩阵内所有元素的总和。

 

输出格式:

每个输出1行,输出每次操作2的结果。

 

代码如下:

  1 #include
  2 #include
  3 #include
  4 #include
  5 #include
  6 //#include
  7 
  8 #include<string>
  9 #include
 10 #include
 11 #include
 12 #include
 13 #include
 14 
 15 using namespace std;
 16 
 17 void read(int &x){
 18     x=0;
 19     char t=getchar();
 20     bool f=0;
 21     
 22     while(t<'0' || t>'9'){
 23         if(t=='-')f=1;
 24         t=getchar();
 25     }
 26     
 27     while(t>='0' && t<='9'){
 28         x=(x<<3)+(x<<1)+t-'0';
 29         t=getchar();
 30     }
 31     
 32     if(f)x=-x;
 33 }
 34 
 35 void add(int,int,int);
 36 int query(int,int);
 37 
 38 int a[1010][1010];
 39 int c[1010][1010];
 40 
 41 int n,m,i,j;
 42 
 43 int T,f,x,y,k;
 44 
 45 int x1,y1,x2,y2;
 46 
 47 int main(){
 48     read(n);
 49     read(m);
 50     
 51     for(i=1;i<=n;i++)
 52         for(j=1;j<=m;j++){
 53             read(a[i][j]);
 54             add(i,j,a[i][j]);
 55         }
 56     
 57     read(T);
 58     
 59     while(T--){
 60         read(f);
 61         
 62         if(f==1){
 63             read(x);read(y);read(k);
 64             add(x,y,k);
 65         }
 66         
 67         if(f==2){
 68             read(x1);read(y1);read(x2);read(y2);
 69             printf("%d\n",query(x2,y2)-query(x2,y1-1)-query(x1-1,y2)+query(x1-1,y1-1));
 70         }
 71     }
 72     
 73     return 0;
 74 }
 75 
 76 void add(int a,int b,int x){
 77     int i=a,j;
 78     
 79     while(i<=n){
 80         j=b;
 81         
 82         while(j<=m){
 83             c[i][j]+=x;
 84             j+=j&-j;
 85         }
 86         
 87         i+=i&-i;
 88     }
 89 }
 90 
 91 int query(int a,int b){
 92     int ans=0,i=a,j;
 93     
 94     while(i>=1){
 95         j=b;
 96         
 97         while(j>=1){
 98             ans+=c[i][j];
 99             j-=j&-j;
100         }
101         
102         i-=i&-i;
103     }
104     
105     return ans;
106 }

 

转载于:https://www.cnblogs.com/running-coder-wfh/p/7201805.html

你可能感兴趣的:([模板]二维树状树组 单点修改 区间查询)