当前位置: 代码迷 >> 综合 >> POJ--1195--二维树状数组--单点更新+平面查询
  详细解决方案

POJ--1195--二维树状数组--单点更新+平面查询

热度:69   发布时间:2023-12-12 06:12:18.0

有一个正方形游戏背包,大小为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));}}
}