题目链接: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;
}
}
}