树状数组基本操作

一、引入

1.lowbit函数

lowbit()操作：非负整数n在二进制表示下最低位1和后面的0构成的数值

int lowbit(int x){
    return x&(-x);
}

2.差分数组b 

二、树状数组

〔manim | 算法 | 数据结构〕 完全理解并深入应用树状数组 | 支持多种动态维护区间操作_哔哩哔哩_bilibilihttps://www.bilibili.com/video/BV1pE41197Qj/?spm_id_from=333.337.search-card.all.click&vd_source=19c498ee530d109d1e2180ce4cdf3ae8

1.t[x]表示以x为根的子树中叶节点的值。 

2.t[x]节点的长度等于lowbit（x）。

3.t[x]节点的父节点为t[x+lowbit(x)]。

4.整棵树深度为log2 n+1；

5.

1.add(int x,int k)操作：将x加上k

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

2.ask(int x)操作:询问x的前缀和

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

单点修改，查询前缀和

int lowbit(int x){
    return x&(-x);
}
void add(int x,int k){
	for (;x<=n;x+=lowbit(x))t[x]+=k;
}
int ask(int x){
	int ans=0;
	for(;x;x-=lowbit(x))ans+=t[x];
	return ans;
}

 单点修改，单点查询

int lowbit(int x){
    return x&(-x);
}
void add(int x,int k){
	for (;x<=n;x+=lowbit(x))t[x]+=k;
}
int ask(int x){
	int ans=0;
	for(;x;x-=lowbit(x))ans+=t[x];
	return ans;
}
//add(x,k);
//ask(x)-ask(x-1);

单点修改，区间查询

int loebit(int x){
	return x&(-x);
}
void add(int x,int k){
	for(;x<=n;x+=lowbit(x))t[x]+=k;
}
int ask(int x){
	int ans=0;
	for(;x;x-=lowbit(x))ans+=t[x];
	return ans;
}
//add(x,k)
//ask(r)-ask(l-1)

#include <iostream>
using namespace std;
typedef long long ll;
const int N = 500010;
int m, n;
int t[N];
int lowbit(int x) {
	return x & (-x);
}
void add(int x, int k) {
	for (; x <= n; x += lowbit(x))
		t[x] += k;
}
int ask(int x) {
	int ans = 0;
	for (; x; x -= lowbit(x))
		ans += t[x];
	return ans;
}
int main() {

	int x, y, p;
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		scanf("%d", &x);
		add(i, x);
	}
	for (int i = 1; i <= m; i++) {
		cin >> p >> x >> y;
		if (p == 1)
			add(x, y);
		if (p == 2)
			cout << ask(y) - ask(x - 1)<<endl;
	}

	return 0;
}

区间修改，单点查询 （引入差分数组b）

int loebit(int x){
	return x&(-x);
}
void add(int x,int k){
	for(;x<=n;x+=lowbit(x))t[x]+=k;
}
int ask(int x){
	int ans=0;
	for(;x;x-=lowbit(x))ans+=t[x];
	return ans;
}
//add(l,k),add(r+1,-k)
//a[x]+ask(x)
//这里引入了差分数组b，差分数组初始化为0，故前缀和为增量

解法1：将t[N]即差分数组b[n]直接初始化为两数之差，进行存储

#include <iostream>
using namespace std;
typedef long long ll;
const int N = 500010;
int m, n;
int t[N];
int lowbit(int x) {
	return x & (-x);
}

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

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

int main() {

	int x, y, z, p;
	cin >> n >> m;
	int last = 0;
	for (int i = 1; i <= n; i++) {
		scanf("%d", &x);
		add(i, x - last);
		last = x;
	}
	for (int i = 1; i <= m; i++) {
		cin >> p ;
		if (p == 1) {
			cin >> x >> y >> z;
			add(x, z);
			add(y + 1, - z);
		}
		if (p == 2){cin >> x;
		cout << ask(x) << endl;
		}
	}
	return 0;
}

 解法2：不初始化t，即将t的值全部初始化为0

#include <iostream>
using namespace std;
typedef long long ll;
const int N = 500010;
int a[N];
//int b[N];
int m, n;
int t[N];

int lowbit(int x) {
	return x & (-x);
}

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

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

int main() {

	int x, y, z, p;
	cin >> n >> m;
	int last = 0;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		//	cin >> x;
//		scanf("%d", &x);
//		add(i, x - last);
//		last = x;
	}
	for (int i = 1; i <= m; i++) {
		cin >> p ;

		if (p == 1) {
			cin >> x >> y >> z;
			add(x, z);
			add(y + 1, - z);
		}

		if (p == 2) {
			cin >> x;
			cout << ask(x) + a[x] << endl;
		}

	}

	return 0;
}

区间修改，区间查询