洛谷 P3374/P3368 【模板】树状数组 1 / 2

【模板】树状数组 1

$Description$

一个长度为$n$的序列$a,$两种操作：
$1.1xy——a_x=y$
$2.2xy——$查询${\sum }_{i=x}^y{a}_i$

$Solution$

单点修改略（
区间查询:$[l,r]$的区间和就是$([1,r]$的区间和$)-($$[1,l-1]$的区间和$)$

$Acceptedcode$

#include<cstdio>

using namespace std;

int n, m;
int t[500005];

void update(int x, int k) {
	for (; x <= n; x += x & -x) t[x] += k;
}

int get_sum(int x) {
	int res = 0;
	for (; x; x -= x & -x) res += t[x];
	return res;
}

int main() {
	scanf("%d %d", &n, &m);
	for (int x, i = 1; i <= n; ++i)
		scanf("%d", &x), update(i, x);
	for (int x, y, ok, i = 1; i <= m; ++i) {
		scanf("%d %d %d", &ok, &x, &y);
		if (ok == 1) update(x, y);
		else printf("%d\n", get_sum(y) - get_sum(x - 1));
	}
}

---

【模板】树状数组 2

$Description$

一个长度为$n$的序列$a,$两种操作：
区间修改，单点查询

$Solution$

树状数组维护$a$的差分数组$d$
$d_i=a_i-a_{i-1}$ 其中$a_0=0$
显然$a_i={\sum }_{j=1}^i{d}_j$
所以单点查询求${\sum }_{i=1}^x{d}_i$
区间修改：
我们研究修改$[l,r]$的值时$d$会怎么变
假设$a=${$1,3,4,7,2$}
则有$d=${$1,2,1,3,-5$}
假设我们给区间$[2,4]$加上$2$
则$a=${$1,5,6,9,2$}$,d=${$1,4,1,3,-7$}
显然当给区间$[l,r]$加$k$时
$d_l=d_l+k,d_{r+1}=d_{r+1}-k$
(易证

$Acceptedcode$

#include<cstdio>

using namespace std;

int n, m;
int t[500005];

void update(int x, int k) {
	for (; x <= n; x += x & -x) t[x] += k;
}

int get_sum(int x) {
	int res = 0;
	for (; x; x -= x & -x) res += t[x];
	return res;
}

int main() {
	scanf("%d %d", &n, &m);
	for (int x, y = 0, i = 1; i <= n; ++i)
		scanf("%d", &x), update(i, x - y), y = x;
	for (int x, y, k, ok, i = 1; i <= m; ++i) {
		scanf("%d %d", &ok, &x);
		if (ok == 1) scanf("%d %d", &y, &k), update(x, k), update(y + 1, -k);
		else printf("%d\n", get_sum(x));
	}
}