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

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

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值