lowbit和树状数组的理解与部分应用

目录

一、前言

二、树状数组

三、题例

1、上链接

2、基本思路

3、代码

（1）C++（AC）

（2）python（7/10  有3个样例时间超限）

---

一、前言

对于学计算机的同学来说，学习算法是一件非常重要的事情，废话不多讲，我们来讲讲“树状数组问题”。

二、树状数组

相关问题：

1、什么是树状数组
顾名思义就是一个结构为树形结构的数组，与二叉树的结构类似但又不同，它是在二叉树的结构上删除了一些中间节点，来看下面这幅图就明白了。

2、树状数组可以解决什么问题呢
可以解决大部分区间上面的修改以及查询的问题，例如：

①单点修改，单点查询

②区间修改，单点查询

③区间查询，区间修改

换言之，线段树能解决的问题，树状数组大部分也可以，但是并不一定都能解决，因为线段树的扩展性比树状数组要强。

3、树状数组和线段树的区别在哪儿
有人会问了既然线段树的问题能够用树状数组解决而且线段树还比树状数组扩展性强，那为什么不直接用线段树呢？问的很好，树状数组的作用就是为了简化线段树

举个例子：一个问题可以用线段树解决写代码半个小时，但是用树状数组只需要10分钟，那么你会选择哪一个算法呢?没错，基于某些简单的问题，我们没必要用到功能性强但实现复杂的线段树(杀鸡焉用宰牛刀)。

4、树状数组的优点
优点：修改和查询操作复杂度于线段树一样都是logN，但是常数比线段树小，并且实现比线段树简单。

缺点：扩展性弱，线段树能解决的问题，树状数组不一定能解决。

5、前置知识—lowbit(x)运算
如何计算一个非负整数n在二进制下的最低为1及其后面的0构成的数？

所以，lowbit的值为n&-n。

教学视频（讲得非常好！！！墙裂推荐！！！）：〔manim | 算法 | 数据结构〕 完全理解并深入应用树状数组 | 支持多种动态维护区间操作_哔哩哔哩_bilibili

（复习的时候去看我的算法笔记“树状数组”部分就行）

三、题例

1、上链接

P3374 【模板】树状数组 1 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

2、基本思路

常规套路树状数组。

3、代码

（1）C++（AC）

#include<iostream>
using namespace std;

int n,m;
const int N=500009;
int a[N],t[N]; //a是待输入的数组，t是树状数组，t[x]保存以x为根的子树中叶节点值的和 

void createTree(){         //构建树状维护数组 t 
	for(int i=1;i<=n;i++){
		int j=i+(i&-i);   //当前节点的父亲 
		if(j<n+1){
			t[j]+=t[i];
		}
	}
}

int lowbits(int x){   //lowbits
	return x&-x;
}

void add(int x,int k){
	for(;x<=n;x+=lowbits(x)) t[x]+=k;
}

int ask(int x){  //
	int ans=0; 
	for(;x;x-=x&-x)
		ans+=t[x];
		return ans;
}

int main(){
	cin>>n>>m; 
	for(int i=1;i<=n;++i){
		cin>>a[i];
		t[i]=a[i];
	}
	createTree();
	while(m--){
		int x,y,z;
		cin>>x>>y>>z;
		if(x==1){
			add(y,z);
		}else if(x==2){
			int ans=ask(z)-ask(y-1);
			cout<<ans<<endl;
		}else{
			cout<<"error"<<endl;
		}
	}
	return 0;
} 

（2）python（7/10  有3个样例时间超限）

n,m=map(int,input().split())
a=[0]*500009 # a是待输入的数组 
t=[0]*500009 # t是树状数组,t[x]保存以x为根的子树中叶节点值的和

def createTree():  # 构建树状维护数组 t 
    global t
    for i in range(1,n+1):
        j=i+(i&-i)   # 当前节点的父亲 
        if j<n+1:
            t[j]+=t[i]

def lowbits(x)->int:  # 没用的函数，只是说x的lowbits就是x&-x
    return x&-x

def add(x,k):
    while(x<=n):
        t[x]+=k
        x+=x&-x

def ask(x)->int:
    ans=0
    while(x>0):
        ans+=t[x]
        x-=x&-x
    return ans

line=[0]+list(map(int,input().split()))
for i in range(1,n+1):
    a[i]=line[i]
    t[i]=line[i]

createTree()

for _ in range(m):
    x,y,z=map(int,input().split())
    if x==1:
        add(y,z)
    elif x==2:
        ans=ask(z)-ask(y-1)
        print(ans)
    else:
        print("error")

以上，树状数组

祝好