有一个正方形游戏背包,大小为S*S(1<=S<=1024)。含有很多小格子,小格子编号从0开始,直到S-1。每一个格子里有一定数量的物品,同时每一个格子的物品里的数目也是不断变化的,现在要一边进行修改某些单位格子内的物品的数目,同时也要询问某些区域的手机数目。 数据保证每个格子内物品数目一直在int范围内,查询的格子内的物品数目也在int范围内
Input
有多组数据,每一行先输入一个数字n,如果n=0再输入S,表示背包的大小,并且初始情况所有的格子内都没有物品:如果n=1,接着输入x,y,a,表示坐标(x,y)的这个小格内增加a个手机:如果n=2,接着输入x1,y1,x2,y2.表示要查询的矩形区域的左上角坐标和右下角坐标,如果n=3,程序结束。
Output
对于每次查询,输出这个区域内的物品数目。
Sample Input
0 4
1 1 2 3
2 0 0 2 2
1 1 1 2
1 1 2 -1
2 1 1 2 3
3
Sample Output
3
4
思路:一维树状数组改进即可;
注意:查询数目:类似于概率论中二维随机变量的分布问题!!!
int x1,x2,y1,y2;
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
printf("%d\n",sum(x2+1,y2+1)-sum(x2+1,y1)-sum(x1,y2+1)+sum(x1,y1));
完整代码:
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#include<cstdio>
#define LL long long
using namespace std;
const int maxa=1024+10;
int a[maxa][maxa];
int n,p;
int lowbit(int i){
return i&-i;
}
void add(int x,int y,int p){
for(int i=x;i<=n;i+=lowbit(i))
for(int j=y;j<=n;j+=lowbit(j))
a[i][j]+=p;
}
int sum(int x,int y){
int ans=0;
for(int i=x;i>0;i-=lowbit(i))
for(int j=y;j>0;j-=lowbit(j))
ans+=a[i][j];
return ans;
}
int main(){
while(scanf("%d",&p)){
if(p==3) break;
else if(p==0){
scanf("%d",&n);
memset(a,0,sizeof a);
}else if(p==1){
int x,y,b;
scanf("%d%d%d",&x,&y,&b);
add(x+1,y+1,b);
}else if(p==2){
int x1,x2,y1,y2;
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
printf("%d\n",sum(x2+1,y2+1)-sum(x2+1,y1)-sum(x1,y2+1)+sum(x1,y1));
}
}
}