树状数组轻度扩展

1.最简单的就不过多赘述，看看别人的文章就行

对于数组a,多次操作加询问，修改其某个点，求其任意区间和

树状数组d[x]保存a的序列区间[x-lowbit(x)+1,x];

二进制k=7=++

即被分为【1,4】【5,6】【7,7】三个小区间

从1~n，每一个数的前缀和都是被如此·安排

单点修改，区间和查询

ll d[N], n,m;//d为树状数组
//低位运算
ll lowbit(ll x) {
	return x & (-x);
}
//求1~x区间的前缀和
ll Sum(ll x) {
	ll res = 0;
		while (x) {
			res += d[x];
			x -= lowbit(x);
		}
		return res;
}
//单点修改
void add(int i, int z) {
	while (i <= n) {
		d[i] += z;
		i += lowbit(i);
	}
}
//区间和查询
ll query(ll l,ll r){
    return Sum(r)-Sum(l-1);
}
int main() {
	cin >> n;
	ll k;
	for (int i = 1; i <= n;i++) {
		cin >> k;
		add(i, k);
	}
	return 0;
}

区间修改，单点查询

此与上面区别在于，此处维护的是数组的差分数组作为树状数组,进而转化成了问题1

1.查询：设原数组为a[i], 设数组d[i]=a[i]−a[i−1](a[0]=0)，则 a[i]=d[j]，可以通过求d[i]的前缀和查询

2.修改：与问题1相同

ll n, a[N], d[N],m;
ll lowbit(ll x) {
	return x & (-x);
}
void add(ll x, ll v) {
	while (x <= n) {
		d[x] += v;
		x += lowbit(x);
	}
}
//使[l,r]区间的值加v
void updata(ll l, ll r, ll v) {
	add(l, v);
	add(r + 1, -v);
}
//单点查询，d的x个前缀和即为x点的值
ll Sum(ll x) {
	ll res = 0;
	while (x) {
		res += d[x];
		x -= lowbit(x);
	}
	return res;
}
int main() {
	cin >> n>>m;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		add(i, a[i] - a[i - 1]);
	}
	while (m--) {
		int a, b, c, d;
		cin >> a;
		if (a == 1) {
			cin >> b >> c >> d;
			updata(b, c, d);
		}
		else {
			cin >> b;
			cout << Sum(b) << "\n";
		}
	}
	return 0;
}

区间修改 + 区间查询

我们基于问题2的“差分”思路，考虑一下如何在问题2构建的树状数组中求前缀和：

位置p的前缀和 =

a[i]=d[j]

在等式最右侧的式子d[j]中，d[1] 被用了p次，d[2]被用了p−1次……那么我们可以写出：

位置p的前缀和

d[j]=d[i]∗(p−i+1)=(p+1)∗d[i] − i∗d[i]

那么我们可以维护两个数组的前缀和：

一个数组是 sum1[i]=d[i]，

另一个数组是 sum2[i]=i∗d[i]。

ll n, sum[N], c[2][N], m;
ll lowbit(ll x) {return x & (-x);}
void add(ll x, ll v,int k) {
	while (x <= n) {
		c[k][x] += v;
		x += lowbit(x);
	}
}
//一次更新范围，需要维护两个数组
void updata(ll l, ll r, ll v) {
    //对sum1的维护
	add(l, v,0);
	add(r + 1, -v,0);
    //对sum2的维护
	add(l, l*v, 1);
	add(r + 1, -(r+1)*v, 1);
}
//求前缀和
ll Sum(ll x,int k) {
	ll res = 0;
	while (x) {
		res += c[k][x];
		x -= lowbit(x);
	}
	return res;
}
int main() {
	cin >> n >> m;
    //构建初始数组前缀和
	for (int i = 1; i <= n; i++) {
		cin >> sum[i];
		sum[i] += sum[i - 1];
	}
    //操作
	while (m--) {
		int  l,r,d;
		char a;
		cin >> a;
		if (a == 'C') {
		//区间修改
            cin >> l >> r >> d;
			updata(l, r, d);
		}
		else {
         //区间查询
			cin >> l >> r;
         //最后的区间修改值是两个树状数组的区间和的差（上面拆分解释）
         //再将其加上原数组的区间和就是修改后的区间和
         //原数组的区间和sum[r]-sum[l-1]
         // sum1的区间和：(r + 1) * Sum(r, 0) - Sum(r, 1)
		 // sum2的区间和：l*Sum(l-1,0)-Sum(l-1,1
            cout << sum[r]-sum[l-1] + (r + 1) * Sum(r, 0) - Sum(r, 1)-(l*Sum(l-1,0)-Sum(l-1,1)) << "\n";
		}
	}
	return 0;
}