树状数组简要总结

1.问题描述

单点修改,区间查询
多了树状数组也干不了,接下来全文描述的是单点加,区间和查询

2.算法原理

树状数组图示
树状数组维护的是[1,n]的区间和,也就是前缀和,实际使用的时候(求[l,r]的区间和)还需要差分一下
也就是:sum(l,r) = sum(1,r) - sum(1,l-1)

现在来看一个家喻户晓的结论:任何一个数都可以化作2的幂的和
用这个思想,比如一个区间长度是7,那么可以分成区间长度4,2,1的三个区间,如果区间的起始位置不确定,那么这样没法给出一个方便的数据结构,但是我们可以利用差分计算区间和,所以可以所以区间的起点都从编号1开始
以7为例:[1,7] = [1,4] + [5,6] + [7,7]
区间长度分别是4,2,1
假如我们换一种区间的分配方式可以吗? [1,7] = [1,1] + [2,3] + [4,7]
区间长度分别是1,2,4

乍一看很合理,但是这样仅对于7合理,如果你想算[1,6]了,会发现没维护[1,2]这个区间的值,所以区间长度2的整数次幂从大到小的顺序进行维护,每个区间如果还可再分,就同样的思想递归(此时你就会发现你递归着建树了),也就是上图中的树状数组,并且这个树的节点个数就是数组的长度,非常地节省空间,时空均比线段树占优势(时间上常数小)

从左往右区间长度是递减的,那么对于给定区间,查询区间和,区间长度每次减小的幅度就是2的幂的和中最小的那一项,称之为“lowbit”,可以巧妙利用计算机二进制的补码表示轻易获得

并且有:对于每个节点x,它的父节点是x+lowbit(x) 因为这样加完之后,恰好x进位

也就是说:无论是单点修改还是区间查询,每次变动幅度都是lowbit(x)

3.代码实现

#include<iostream>
using namespace std;
int sum[500001],n,m;

//求二进制下最末一个1
int lowbit(int x)
{
	return x&(-x);
}

//单点加
void add(int idx,int x)
{
	for(int i=idx;i<=n;i+=lowbit(i))
	sum[i]+=x;
}

//[1,x]和查询
int query_sum(int x)
{
	int ans=0;
	for(int i=x;i>=1;i-=lowbit(i))
	ans+=sum[i];
	return ans;
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		int x;
		cin>>x;
		add(i,x);
	}
	for(int i=1;i<=m;i++)
	{
		int order;
		cin>>order;
		if(order==1)
		{
			int x,k;
			cin>>x>>k;
			add(x,k);
		}
		else{
			int x,y;
			cin>>x>>y;
			cout<<(query_sum(y)-query_sum(x-1))<<endl;
		}
	}
	return 0;
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值