PKU1195 Mobile phones - 二维树状数组

题目描述:

在一个N*N的区域内,可以在某个点增加或减少A,并动态询问一个矩形区域内的数值之和。(N<=1024)

分析:

典型的二维树状数组的应用。

以前有几篇论文讨论了如何要用二维线段树来做,相当麻烦。不过二维树状数组则十分简单。

不过树状数组的应用范围很窄,如果修改题目为“可以给某个矩形区域每个点增加或减少A,并动态询问某个点的数值”那么树状数组就不行了,而线段树可以使用。

  1. /*
  2. PKU1195 Mobile phones
  3. */
  4. #include <stdio.h>
  5. #include <memory.h>
  6. #define clr(a) memset(a,0,sizeof(a))
  7. #define N 1030
  8. int c[N][N];
  9. int n;
  10. int lowbit(int x){
  11.     return x&(-x);
  12. }
  13. void Update(int x,int y,int t){
  14.     int i,j;
  15.     i=x;
  16.     while(i<=n){
  17.         j=y;
  18.         while(j<=n){
  19.             c[i][j]+=t;
  20.             j+=lowbit(j);
  21.         }
  22.         i+=lowbit(i);
  23.     }
  24. }
  25. int Query(int x,int y){
  26.     int i,j,s=0;
  27.     i=x;
  28.     while(i>0){
  29.         j=y;
  30.         while(j>0){
  31.             s+=c[i][j];
  32.             j-=lowbit(j);
  33.         }
  34.         i-=lowbit(i);
  35.     }
  36.     return s;
  37. }
  38. int main()
  39. {
  40.     int i,j,k;
  41.     int x,y,xt,yt,t;
  42.     
  43.     scanf("%*d%d",&n);
  44.     while(scanf("%d",&k)!=EOF && k!=3){
  45.         if(k==1){   //Update
  46.             scanf("%d%d%d",&x,&y,&t);
  47.             x++;y++;
  48.             Update(x,y,t);
  49.         }
  50.         if(k==2){   //Query
  51.             scanf("%d%d%d%d",&x,&y,&xt,&yt);
  52.             x++;y++;xt++;yt++;
  53.             t=Query(xt,yt)-Query(x-1,yt)-Query(xt,y-1)+Query(x-1,y-1);
  54.             printf("%d/n",t);
  55.         }
  56.     }
  57.     
  58.     return 0;
  59. }

你可能感兴趣的:(PKU1195 Mobile phones - 二维树状数组)