[浅谈数据结构] 浅谈树状数组

1.作用

树状数组是一种高效 而简单的数据结构,用于*大部分区间修改查询问题 ,形如\(a[1]+a[2]+a[3]+a[4]+...+a[n]\)(其不支持的可以由线段树替代)

2.选择原因

优点:树状数组的码量 明显比线段树时间复杂度 比朴素算法与线段树更空间复杂度吊打线段树

缺点:部分线段树能解决的问题树状数组解决不了

3.基本原理&实现方法

如图(from OIWiki)

在求解\(a[1]+a[2]+a[3]+a[4]+...+a[n]\)这类问题时,根据上图这种数据结构,我们可以高效的进行查询

3.0.lowbit

3.0.1.思路

干什么的 :求一个非负整数\(n\)在二进制下的最低为1的位的个数(1)及其后面的0的位数构成的数(1+后面0的个数)

怎么干 :return x&(-x)

原理(不会可以跳过,但要背结论):

我们得到lowbit的值,只需要得到最后一个1的位置,并且把除了这个位置之外的所有位置全部置成零。然后输出就可以。思路有了,如何操作?

根据计算机补码的性质,补码就是原码的反码加一

如:

\((110)_{2}\)

反码:

\((001)_{2}\)

加一:

\((010)_{2}\)

可以发现变为反码后 \(x\) 与反码数字位每一位都不同, 所以当反码加\(1\)后会逢\(1\)一直进位直到遇到\(0\),这个\(0\)变成了\(1\),操作停止。

进位的部分相当于再一次取反,也就还原原著。而最后变为1的部分又停在最后一个为0的位置,也就是取反前1的位置了,正好完成操作

又有人要问了:主播主播,我也没得到最终结果啊,如果停止进位后前面还有0呢?操作前二进制后的数我们默认它不存在前导零,也就是最高位不可能是0,也就必定是1,取反后为0,得到的一定是一最高位为0的数,而0又可以舍去,因此合理

举个例子,110000~001111 形如0001000可以转化为1000,也就是说最高位不可能是0

3.0.2.代码

cpp 复制代码
int lowbit(int x)
{
	return x&(-x);
}

3.1.如何修改单点?

3.1.1.思路

更新一个点也要更新其祖上十八代,祖上十八代怎么推?

我们发现每向上一层\(lowbit\)值都增加\(1\),因此得到增加单点代码:

3.1.2 代码

修改单点:

cpp 复制代码
void build(int x,int k)
{
	for(int i=x;i<=n;i+=lowbit(i))
	{
		c[i]+=k;
	}
}

修改单点的扩大既是建树(只进行浅谈篇的操作):

cpp 复制代码
void build(int x,int k)
{
	for(int i=x;i<=n;i+=lowbit(i))
	{
		c[i]+=k;
	}
}
//...
int main()
{
	for (int i = 1; i <= n; i++) 
	{
		add(i, c[i]);
	}
}

3.2.如何查询1~x的和?

3.2.1.思路

举例计算\(a[1]+a[2]+a[3]+a[4]+a[5]+a[6]+a[7]\)的和

从\(a[7]\)开始跳,跳到\(c[7]\)上,发现\(c[7]\)只管辖\(a[7]\),再跳到\(c[6]\)上,发现其管辖\(a[5]+a[6]\),再跳到\(a[4]\)上,发现其管辖\(a[1]+a[2]+a[3]+a[4]\),发现我们得到最终答案。

完整推导:

\(a[1]+a[2]+a[3]+a[4]+a[5]+a[6]+a[7]=c[7]+c[6]+c[4]\)

关注等式右侧三个数在树上的关系

再关注等式右侧本身的关系,先推导出\(4,6,7\)的二进制表示

\(4=11_{2},6=110_{2},7=111_{2}\)

研究其关系,发现\(6=7-lowbit(7),4=6-lowbit(6)\)

所以\(code\):

3.2.2.代码

cpp 复制代码
int ask(int x)
{
	int ans=0;
	for(int i=1;i<=x;i-=lowbit(x))
	{
		ans+=c[i];
	}
	return ans;
}

3.3.如何查询任意数~x的和?

3.3.1.思路

前缀和相减(听着很抽象)

公式:\(a[1,r]-a[1,l-1]=a[l,r]\)

3.3.2.代码

cpp 复制代码
int ask(int x)
{
	int ans=0;
	for(int i=L-1;i;i-=lowbit(i))
	{
		ans-=c[i];
	}
	for(int i=R;i;i-=lowbit(i))
	{
		ans+=c[i];
	}
	return ans;
}

4.总结&练习&展望

4.1.总结

在浅谈篇中,注意到我们使用树状数组进行了查询区间和修改单点 的操作。这是最基本的使用。回忆一下,该二操作关键点在于\(lowbit\)。

4.2.练习

建议同学们完成:

模板:树状数组1

找逆序对

4.3.展望

在下一篇再谈篇中,我们将学习使用前缀和与差分实现区间修改与单点查询,~~~然后就可以开YNOI毒瘤了~

注意到本文由博客园 @OIRikka,洛谷 @March7thDev撰写,禁止任何形式的转载!