ybtoj树状数组课堂过关二维树状数组 例题5单点修改区间查询

Posted 2021-06-19 SSL_ZZL

【例题5】单点修改区间查询

• Link
• 题目大意
• 解题思路
• Code

---

Link

ybtoj【树状数组课堂过关】 【例题5】单点修改区间查询
题面//因为不知道侵不侵权所以就是题面是私密的，有账号的直接看转送门就可了

---

题目大意

给一个矩阵A，有两个操作

• 1 a, b, c =》$A_{a,b}$+c
• 2 a, b, c, d =》求矩阵$A_{(a到c),(b到d)}$的和

---

解题思路

从一维树状数组到二维树状数组
其实差不多就是把循环变为两层啦…

求和的时候要容斥一下

---

Code

#include <iostream>
#include <cstdio>
#define ll long long

using namespace std;

int n, m, x, y, xx, yy, c;
ll s, tree[5000][5000];

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

void add(int x, int y, ll s) {
	for(; x <= n; x += lowbit(x))
		for(int now = y; now <= m; now += lowbit(now))  //每层的y都从'y'开始，直接使用y的话，第二层就是'm'了 TAT
			tree[x][now] += s;
}

ll sum(int x,int y) {
	ll ans = 0;
	for(; x; x -= lowbit(x))
		for(int now = y; now; now -= lowbit(now))  //同上
			ans += tree[x][now];
	return ans;
}

int main() {
	scanf("%d %d", &n, &m);
	while(scanf("%d", &c) != EOF) {
		if(c == 1) {
			scanf("%d %d %lld", &x, &y, &s);
			add(x, y, s);
		} else {
			scanf("%d %d %d %d", &x, &y, &xx, &yy);
			printf("%lld\\n", sum(xx, yy) - sum(x - 1, yy) - sum(xx, y - 1) + sum(x - 1, y - 1));  //容斥
		}
	}
}