POJ1195 Mobile phones 二维树状数组

来源:互联网 发布:手机直播软件大全 编辑:程序博客网 时间:2024/05/16 08:22

题目链接:http://poj.org/problem?id=1195


题目大意:给出一个n*n的矩阵,两种操作:“1 x y v”表示将矩阵上的点(x,y)加上v;“2 x1 y1 x y”表示查询矩阵的和。


分析:和HDU2645 Stars类似,很裸的二维树状数组的题。


实现代码如下:

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int M=1100;int a[M][M],n;int lowbit(int i){    return i&(-i);}void update(int x,int y,int v){    for(int i=x;i<M;i+=lowbit(i))      for(int j=y;j<M;j+=lowbit(j))        a[i][j]+=v;}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+=a[i][j];    return sum;}int main(){    int op,x,y,v,x1,y1;    while(true)    {        scanf("%d",&op);        switch(op)        {            case 0:scanf("%d",&n);                   memset(a,0,sizeof(a));                   break;            case 1:scanf("%d%d%d",&x,&y,&v);                   update(x+1,y+1,v);                   break;            case 2:scanf("%d%d%d%d",&x1,&y1,&x,&y);                   x++; y++; x1++; y1++;                   printf("%d\n",query(x,y)-query(x,y1-1)-query(x1-1,y)+query(x1-1,y1-1));                   break;            case 3:return 0;        }    }}


0 0
原创粉丝点击