p1358 打鼹鼠(树状数组——二维BIT)

题目

背景 Background
SuperBrother在机房里闲着没事干(再对比一下他的NOIP,真是讽刺啊…),于是便无聊地开始玩“打鼹鼠”…
描述 Description
  在这个“打鼹鼠”的游戏中,鼹鼠会不时地从洞中钻出来,不过不会从洞口钻进去(鼹鼠真胆大……)。洞口都在一个大小为n(n<=1024)的正方形中。这个正方形在一个平面直角坐标系中,左下角为(0,0),右上角为(n-1,n-1)。洞口所在的位置都是整点,就是横纵坐标都为整数的点。而SuperBrother也不时地会想知道某一个范围的鼹鼠总数。这就是你的任务。
输入格式 Input Format
每个输入文件有多行。
第一行,一个数n,表示鼹鼠的范围。
以后每一行开头都有一个数m,表示不同的操作:
m=1,那么后面跟着3个数x,y,k(0<=x,y m=2,那么后面跟着4个数x1,y1,x2,y2(0<=x1<=x2 m=3,表示老师来了,不能玩了。保证这个数会在输入的最后一行。
询问数不会超过10000,鼹鼠数不会超过maxlongint。
输出格式 Output Format
对于每个m=2,输出一行数,这行数只有一个数,即所询问的区域内鼹鼠的个数。
样例输入 Sample Input
4
1 2 2 5
2 0 0 2 3
3
样例输出 Sample Output
5
时间限制 Time Limitation
各个测试点1s
注释 Hint
水题一道。
所有数据均为随机生成,包括样例……
来源 Source
gnaggnoyil、

题解

本题是一道二维树状数组。(好像是道板子题

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1050;
inline ll read()
{
	ll f=1,num=0;
	char ch=getchar();
	while (!isdigit(ch)) { if (ch=='-') f=-1; ch=getchar(); }
	while (isdigit(ch)) num=(num<<1)+(num<<3)+(ch^48), ch=getchar();
	return num*f;
}
ll n,m,c[maxn][maxn];
ll lowbit(ll x)
{
	return x & -x;
}
void add(ll x,ll y,ll z)
{
	for (int i=x;i<=n;i+=lowbit(i))
		for (int j=y;j<=n;j+=lowbit(j))
			c[i][j]+=z;
}
ll sum(ll x,ll y)
{
	ll ans=0;
	for (int i=x;i>=1;i-=lowbit(i))
		for (int j=y;j>=1;j-=lowbit(j))
			ans+=c[i][j];
	return ans;
}
int main()
{
	n=read();
	while (1)
	{
		m=read();
		if (m==1)
		{
			ll x=read(),y=read(),k=read();
			++x,++y;
			add(x,y,k);
		}
		else if (m==2)
		{
			ll x=read(),y=read(),a=read(),b=read();
			++x,++y,++a,++b;
			printf("%lld\n",abs(sum(a,b)-sum(x-1,b)-sum(a,y-1)+sum(x-1,y-1)));
		}
		else return 0;
	}
}

你可能感兴趣的:(树状数组)