树状数组

/*
TreeArray.h
树状数组,一维和二维都有。数组必须从1开始

问题:
已知数组a[],元素个数为n,现在更改a中的元素,要求得新的a数组中i到j区间内的和

解决方法:
从图中不难发现,c[k]存储的实际上是从k开始向前数k的二进制表示中右边第一个1所代表的数字
个元素的和(这么说可能有点拗口,令lowbit为k的二进制表示中右边第一个1所代表的数字,然后
c[k]里存的就是从a[k]开始向前数lowbit个元素之和)
C1 = A1
C2 = A1 + A2
C3 = A3
C4 = A1 + A2 + A3 + A4
C5 = A5
C6 = A5 + A6
C7 = A7
C8 = A1 + A2 + A3 + A4 + A5 + A6 + A7 + A8

这么存的好处:
无论是树状数组还是线段树,都用到了分块的思想
方便计算,我们可以用位运算轻松地算出lowbit.

时间复杂度:
对于更改元素来说,如果第i个元素被修改了,可以直接在c数组里面进行相应的更改,如图中的例子,
假设更改的元素是a[2],那么它影响到得c数组中的元素只有c[2],c[4],c[8],我们只需一层一层往
上修改就可以了,这个过程的最坏的复杂度也不过O(logN);
对于查找来说,如查找s[k],只需查找k的二进制表示中1的个数次就能得到最终结果,比如查找s[7],7的二进制表示中有3个1,也就是要查
找3次,到底是不是呢,我们来看上图,s[7]=c[7]+c[6]+c[4]

怎么实现这个过程:
还以7为例,二进制为0111,右边第一个1出现在第0位上,也就是说要从a[7]开始向前数1个元素(只
有a[7]),即c[7];
然后将这个1舍掉,得到6,二进制表示为0110,右边第一个1出现在第1位上,也就是说要从a[6]开始
向前数2个元素(a[6],a[5]),即c[6];
然后舍掉用过的1,得到4,二进制表示为0100,右边第一个1出现在第2位上,也就是说要从a[4]开始
向前数4个元素(a[4],a[3],a[2],a[1]),即c[4].
*/
#include<iostream>  
using namespace std; 

#define MAX 1002

class TreeArray
{
public:
	int **s;
	int type;
public:
	TreeArray(int t);
	~TreeArray();
	void clear();
	int lowbit(int x){return x&(-x);};
	void modify(int x, int value);
	void modify(int x, int y, int value);
	int sum(int x);
	int sum(int x, int y);
};
TreeArray::TreeArray(int t):type(t)
{
	int i;
	s = new int*[MAX+1];
	//一维
	if(type == 1)
	{
		for(i = 0; i <= MAX; i++)
			s[i] = new int;
	}
	//二维
	else if(type == 2)
	{
		for(i = 0; i <= MAX; i++)
			s[i] = new int[MAX+1];
	}
}
TreeArray::~TreeArray()
{
	int i;
	for(i = 0; i <= MAX; i++)
		delete []s[i];
	delete []s;
}
void TreeArray::clear()
{
	int i, j;
	for(i = 0; i <= MAX; i++)
	{
		if(type == 1)
			s[i][0] = 0;
		else
		{
			for(j = 0; j <= MAX; j++)
				s[i][j] = 0;
		}
	}
}
void TreeArray::modify(int x, int value)
{
	while(x <= MAX)
	{
		s[x][0] += value;
		x += lowbit(x);
	}
}
void TreeArray::modify(int x,int y,int value)
{  
	int temp = y;
	while(x <= MAX)
	{
		y = temp;
		while(y <= MAX)
		{
			s[x][y] += value;
			y = y + lowbit(y);
		}
		x = x + lowbit(x);
    }     
}  
int TreeArray::sum(int x)
{
	int ans=0;
	while(x > 0)
	{
		ans += s[x][0];
		x -= lowbit(x);
	}
	return ans;
}
int TreeArray::sum(int x,int y)
{  
    int ans=0, temp = y;  
	while(x > 0)
	{
		y = temp;
		while(y > 0)
		{
			ans += s[x][y];
			y = y - lowbit(y);
		}
		x = x - lowbit(x);
	}
    return ans;  
}


转载于:https://my.oschina.net/windmissing/blog/690477

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值