二维树状数组

原创 2016年08月30日 21:49:16

刷cf的时候做到一道题,让求矩阵中任意一个子矩阵的元素之和,当时看了一下别人的题解,发现跟树状数组有点像,然后查了一下果然是树状数组,不过是二维的。这里就总结一下二维树状数组的过程。

问题:一个由数字构成的大矩阵,能进行两种操作
1) 对矩阵里的某个数加上一个整数(可正可负)
2) 查询某个子矩阵里所有数字的和,要求对每次查询,输出结果。

求任意子矩阵可以由sum(x2, y2) - sum(x1 - 1, y2) - sum(x2, y1 - 1) + sum(x1 - 1, y1 - 1)得到,这个证明我是不会的,但验证了一下是正确的,记住能用就行。注意树状数组的下标不能从0开始,必须从1开始。

下面是POJ1195的代码,一道二维树状数组的裸题。

16039476 Seasonal 1195 Accepted 4384K 485MS C++ 896B 2016-08-30 21:42:06

#include<iostream>
#include<algorithm>
using namespace std;
int n;
int map[1050][1050];

int lowbit(int x)//求出最低位的1
{
	return x&(-x);
}
void add(int x,int y,int w)
{
	for (int i = x; i <= n; i += lowbit(i))
		for (int j = y; j <= n; j += lowbit(j))
			map[i][j] += w;
}
int sum(int x, int y)
{
	int temp = 0;
	for (int i = x; i > 0; i -= lowbit(i))
		for (int j = y; j > 0; j -= lowbit(j))
			temp += map[i][j];
	return temp;
}
int getsum(int x1, int y1, int x2, int y2)//可求任意子矩阵,证明不会,但验证了一下是正确的
{
	return sum(x2, y2) - sum(x1 - 1, y2) - sum(x2, y1 - 1) + sum(x1 - 1, y1 - 1);
}

int main()
{
	int pos;
	while (scanf("%d", &pos) && pos != 3)
	{
		if (pos == 0)
		{
			scanf("%d", &n);
		}
		else if (pos == 1)
		{
			int x, y, w;
			scanf("%d%d%d", &x, &y, &w);
			add(x+1, y+1, w);
		}
		else
		{
			int x1, y1, x2, y2;
			scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
			printf("%d\n", getsum(x1+1,y1+1,x2+1,y2+1));
		}
	}
	return 0;
}


版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

一维和二维树状数组的实现

  • 2008年09月15日 15:39
  • 509B
  • 下载

二维树状数组

  • 2013年06月11日 10:48
  • 21KB
  • 下载

poj1656 二维树状数组

/** * poj1656 二维树状数组 * 二维线段树的做法太繁琐了,暂时先不看了,二维树状数组的做法相对简单些,但题目中这样的实现,确实减轻了查询的复杂度,但是插入的复杂度却确实提升了比较多 ...

二维树状数组ppt

  • 2013年06月18日 13:29
  • 167KB
  • 下载

POJ 2155 Matrix (二维树状数组)

链接:http://poj.org/problem?id=2155 Description Given an N*N matrix A, whose elements are either 0...

HDU 2642(树状数组模板二维)

Stars Time Limit: 5000/2000 MS (Java/Others)    Memory Limit: 32768/65536 K (Java/Others) Total ...
  • a73265
  • a73265
  • 2015年08月03日 18:12
  • 204

bzoj3132 二维树状数组

3132: 上帝造题的七分钟 Time Limit: 20 Sec  Memory Limit: 128 MB Submit: 522  Solved: 242 [Submit][Status]...

Matrix - POJ 2155 二维树状数组

Matrix Time Limit: 3000MS   Memory Limit: 65536K Total Submissions: 17150   Accept...

一维 + 二维树状数组 + 单点更新 + 区间更新 详解

树状数组详解: 假设一维数组为A[i](i=1,2,...n),则与它对应的树状数组C[i](i=1,2,...n)是这样定义的:  C1 = A1  C2 = A1 + A2  C3 = A3  C...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:二维树状数组
举报原因:
原因补充:

(最多只允许输入30个字)