题目大意:
有一个S*S的数组。
有两种操作:
一、h[X,Y]加上A
二、查询所有x1<=i<=x2, y1<=j<=y2的h[i, j]的和。
这一题是一道二维树状数组的模板体。
如果要查询(x1,y1,x2,y2)的值:只要查询query(1, 1, x2, y2) + query(1, 1, x1-1, y1-1) - query(1, 1, x2, y1-1) - query(1, 1, x1-1, y2)
这个查询和维护可以使用二维树状数组来维护
我的代码:
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <cstring>
using namespace std;
const int MAXN = 1050;
int h[1050][1050];
int n;
int lowbit(int x)
{
return x & (-x);
}
void updata(int x, int y, int val)
{
int i, j;
for (i=x; i<=n; i+=lowbit(i))
for (j=y; j<=n; j+=lowbit(j))
h[i][j] += val;
}
int query(int x, int y)
{
int i, j, ans = 0;
for (i=x; i>0; i-=lowbit(i))
for (j=y; j>0; j-=lowbit(j))
ans += h[i][j];
return ans;
}
int main()
{
int tt, x, y, a, x1, y1, x2, y2, ans;
scanf("%d%d",&tt,&n);
memset(h, 0, sizeof(h));
while (scanf("%d",&tt))
{
if (tt == 1)
{
scanf("%d%d%d",&x,&y,&a);
x++; y++;
updata(x, y, a);
}
else if (tt == 2)
{
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
x2++; y2++; x1++; y1++;
ans = query(x2,y2) + query(x1-1, y1-1) - query(x2, y1-1) - query(x1-1, y2);
printf("%d\n", ans);
}
else break;
}
return 0;
}