POJ 1195 Mobile phones (二维树状数组, 模板题)

二维树状数组
题目意思:
题意:给出一个S * S的矩阵,每个位置可以放手机,初始时每个位置的都是空的。
现要执行一些操作:1 X Y A,在位置(X, Y)增加A台手机,
2 L B R T,查询(L, B)到(R, T)的矩阵的手机有多少

本题要点:
1、二维树状数组定义:
一维树状数组很容易扩展到二维,在二维情况下:数组A[][]的树状数组定义为:
C[x][y] = ∑ a[i][j], 其中,
x-lowbit(x) + 1 <= i <= x,
y-lowbit(y) + 1 <= j <= y.
然后,只需要把一维的树状数组的 update 函数和getSum 函数改为 二维即可。、
参考网址: https://blog.csdn.net/zzti_xiaowei/article/details/81053094

2、 本题的坐标从0开始,为了方便计算,改为从1 开始.
3、 计算某个子矩阵中所有数的和(横坐标 x1 <= x <= x2, 纵坐标 y1 <= y <= y2),使用的相当于二维前缀和的方法:
子矩阵的和 = getSum(x2, y2) - getSum(x1 - 1, y2) - getSum(x2, y1 - 1) + getSum(x1 - 1, y1 - 1)

#include <cstdio>
#include <cstring>
#include <iostream>
#define lowbit(i) ((i)&(-i))
using namespace std;
const int MaxN = 1030;
int a[MaxN][MaxN];
int c[MaxN][MaxN];	// 二维树状数组
int cmd, n;

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

int getSum(int i, int j)
{
	int sum = 0;
	for(int x = i; x; x -= lowbit(x))
	{
		for(int y = j; y > 0; y -= lowbit(y))
		{
			sum += c[x][y];
		}
	}
	return sum;
}

int main()
{
	scanf("%d%d", &cmd, &n);
	int x, y, d;
	int x1, x2, y1, y2;
	while(scanf("%d", &cmd) != EOF && cmd != 3)
	{
		if(1 == cmd)	//增加
		{	
			scanf("%d%d%d", &x, &y, &d);
			update(x + 1, y + 1, d);
		}else if(2 == cmd){		//查询
			scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
			x1++, y1++, x2++, y2++;
			printf("%d\n", getSum(x2, y2) - getSum(x1 - 1, y2) - getSum(x2, y1 - 1) + getSum(x1 - 1, y1 - 1));
		}
	}
	return 0;
}

/*
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
*/

/*
3
4
*/

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值