关闭

poj 1195 Mobile phones (二维树状数组)

标签: 树状数组
245人阅读 评论(0) 收藏 举报
分类:

题目:http://poj.org/problem?id=1195

题意:给你N*N的矩阵,修改点,查矩阵的和。

分析:很裸的题了。。。

代码:

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

const int maxn =  1048;
int tree[maxn][maxn],lowbit[maxn];

void Init()
{
	for(int i=1;i<maxn;i++)
		lowbit[i]=(i&-i);
}

void update(int x,int y,int c)
{
	for(int i=x;i<maxn && i;i+=lowbit[i])
		for(int j=y;j<maxn && j;j+=lowbit[j])
			tree[i][j]+=c;
}

int query(int x,int y)
{
	int ret(0);
	for(int i=x;i>0;i-=lowbit[i])
		for(int j=y;j>0;j-=lowbit[j])
			ret+=tree[i][j];
	return ret;
}

int main()
{
	int N,q,R,T,L,B,x,y,v;
	Init();
	while(1)
	{
		scanf("%d",&q); 
		if(q==1)
		{
			scanf("%d%d%d",&x,&y,&v);
			x++,y++;
			update(x,y,v);
		}
		else if(q==2)
		{
			scanf("%d%d%d%d",&L,&B,&R,&T);
			L++,B++,R++,T++;
			printf("%d\n",query(R,T)-query(L-1,T)-query(R,B-1)+query(L-1,B-1));
		}
		else if(q==3)
			return 0;
		else
		{
			scanf("%d",&N);
			for(int i=1;i<=N+3;i++)
				memset(tree[i],0,sizeof(tree[0][0])*(N+3));
		}
	}
	return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:69560次
    • 积分:2643
    • 等级:
    • 排名:第13822名
    • 原创:201篇
    • 转载:2篇
    • 译文:0篇
    • 评论:5条
    最新评论