树状数组入门(单点修改+区间查询)

问题的引入:

例:在一个【1,1000000000】的区间上改变一个位置上的值,并且求n次【x,y】内(共n个元素)的区间和;

正常思路(反正我就这么做T-T):更改进行1次操作,时间复杂度O(1),查询n次操作,时间复杂度O(n),并且n次查询区间和,所以时间复杂度O(n^2);           TLE了~~~

那怎么快点解决这个问题呢?没错,用树状数组;

单点修改+区间查询:

                (ps:图是别人那儿拷的,如有侵权请联系我)

上面的图很好的展现了什么是树状数组,但究竟是啥意思呢,各位看官,且听我慢慢道来~~~~

先介绍一个函数:

lowbit函数

顾名思义,lowbit这个函数的功能就是求某一个数的二进制表示中最低的一位1,举个例子,x = 6,它的二进制为110,那么lowbit(x)就返回2,因为最后一位1表示2

那么怎么求lowbit呢?

  把这个数的二进制写出来,然后从右向左找到第一个1(这个1就是我们要求的结果,但是现在表示不出来,后来的操作就是让这个1能表示出来),这个1不要动和这个1右边的二进制不变,左边的二进制依次取反,这样就求出的一个数的补码,说这个方法主要是让我们理解一个负数的补码在二进制上的特征,然后我们把这个负数对应的正数与该负数与运算一下,由于这个1的左边的二进制与正数的原码对应的部分是相反的,所以相与一定都为0,;由于这个1和这个1右边的二进制都是不变的,因此,相与后还是原来的样子,故,这样搞出来的结果就是lowbit(x)的结果。 

  知道了lowbit函数再看我们要什么,我们要做的是,快速的进行【x,y】的区间查询,为什么朴素的厉遍法慢,就是因为它要厉遍【x,y】内每一个元素,但是观察树状数组,不管想求哪个位置上的值我们都只需要进行logn次的操作;

例如:

查询:

7=(111)2,通过lowbit分解,它可以变成3个数的和:(111)2=(1)2+(10)2+(100)2,然后我们分析这个倒着跳的过程。7减去7的最小2次幂2^0得到6。6减去6的最小2次幂2^1得到4。4减去4的最小2次幂2^2得到0;

   sum[7](a【1,7】的和)=c[7]+c[6]+c[4];(也可以叫做a[7]的前缀和)

同理:【x,y】的区间和=sum【y】-sum【x-1】;(因为是二进制的操作,所以查询区间和的操作不会超过logn次,既成功的把时间复杂度降低到了O(logn));

修改:

 一开始我们只有原数组,怎么搭建起树状数组呢?

 其实很简单,可以先把树状数组中的值都想成是0,然后依次处理读入每个a数组中的值,并且注意不仅要要修改当前点的值,还要修改之后父节点的值,因为之后的每次的查询,我们都是算区间端点的前缀和的差;那怎么修改父节点的值呢?刚才我们查询的时候是从后往前找父节点,并且累加,这次我们要从下往上找父节点,并且将父节点也加上修改的值;

这么说可能有点玄学,举个具体的例子,比如在刚才的数组中,我们让a【1】+5,那么不仅a【1】的值变了而且之后每个数的前缀和都变了,(注意是前缀和变了,不是a[i]的值变了)既:a【2】+1,a【4】+1,a【8】+1,再看sum【7】原本=  c[7]+c[6]+c[4],因为 只有c[4]+1,所以sum【7】+1;

说了这么多,总结一下经验树状数组的理解一定要结合我上面给出的图进行理解!     
附上一个测试的代码:

#include <iostream>
#include<algorithm>
#include<cmath>
#include<math.h>
#include<stdio.h>
#include<string>
#include<cstring>
#define maxn  1000000+10
#define pi acos(-1.0)
#define esp 1e-7
typedef long long ll;
using namespace std;

/*树状数组单点更新+区间查询*/
int c[maxn];//树状数组
int a[maxn] = {0,1,2,3,4,5};//原来的普通数组
int n=5;//数组的长度
void update(int x,int val)//更新函数
{
	for (; x <= n; x += (x&-x))//一路找x的父节点,并且更新父节点
	{
		c[x] += val;
	}
}
int ask(int x)//查询前缀和
{
	int res=0;
	for (; x >= 1; x -= (x&-x))//一路向前找之前的父节点,并且加上值;例:15=(1111)2=(1)2+(10)2+(100)2+(1000)2
	{
		res += c[x];
	}
	return res;
}
int query_ask(int l, int r)//区间查询
{
	return ask(r) - ask(l - 1);
}

int main()
{
	int p;
	int flag;
	for (int i = 1; i <= n; i++)//初始化树状数组
	{
		update(i, a[i]);
	}
	for (int i = 1; i <= n; i++)
	{
		cout << c[i] << " ";
	}
	cout << endl;
	system("pause");

	return 0;


}

                                              

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
树状数组(Fenwick Tree)是一种用于高效处理区间修改单点查询的数据结构。下面我将介绍如何使用树状数组实现区间修改单点查询。 首先,我们需要定义树状数组的数据结构。树状数组由一个数组和一组操作组成,其中数组用于存储数据,操作用于更新和查询数组中的。 下面是一个示例的树状数组实现: ```python class FenwickTree: def __init__(self, n): self.size = n self.tree = [0] * (n + 1) def update(self, idx, delta): while idx <= self.size: self.tree[idx] += delta idx += idx & -idx def query(self, idx): res = 0 while idx > 0: res += self.tree[idx] idx -= idx & -idx return res def range_update(self, l, r, delta): self.update(l, delta) self.update(r + 1, -delta) def point_query(self, idx): return self.query(idx) ``` 在上面的代码中,我们定义了一个FenwickTree类,通过构造函数`__init__`来初始化树状数组,并使用`size`来表示数组的大小。`tree`数组用于存储数据。`update`方法用于更新指定位置的,`query`方法用于查询指定位置之前的求和结果。`range_update`方法用于对指定区间进行修改,`point_query`方法用于查询单个位置的。 下面是一个示例的使用场景: ```python # 示例使用 n = 10 tree = FenwickTree(n) # 区间修改 [2, 6] 的加 3 tree.range_update(2, 6, 3) # 查询位置 5 的 value = tree.point_query(5) print(value) ``` 在上面的示例中,我们创建了一个大小为10的树状数组,并对区间[2, 6]的进行了修改,将其加3。然后,我们查询了位置5的,结果为3。 希望以上内容能够帮助到你实现树状数组区间修改单点查询。如果还有其他问题,请随时提问。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值