【POJ 1195】Mobile phones(二维树状数组)

传送门

就是一个二维树状数组单点修改+区间查询的模板题,代码特别好实现,可以来练练手。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int trash,n,tree[1050][1050]; 
inline int lowbit(int x)
{
	return x&(-x);
}
inline void update(int x,int y,int delta)
{
	for(int i=x;i<=n;i+=lowbit(i))
	{
		for(int j=y;j<=n;j+=lowbit(j))
		{
			tree[i][j]+=delta; 
		}
	}
}
inline 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+=tree[i][j];
		}
	}
	return sum;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	while(cin>>trash>>n)
	{
		memset(tree,0,sizeof(tree));
		while(1)
		{
			int opt,a,b,c,d,k;
			cin>>opt;
			if(opt==3)	break;
			if(opt==1)	//单点修改
			{
				cin>>a>>b>>k;
				update(a+1,b+1,k);
			} 
			if(opt==2)
			{
				cin>>a>>b>>c>>d;
				a++,b++,c++,d++;
				cout<<query(c,d)-query(c,b-1)-query(a-1,d)+query(a-1,b-1)<<'\n';
			}
		}
	}
	return 0;
}
posted @ 2018-08-05 22:48  Patrickpwq  阅读(141)  评论(0编辑  收藏  举报