模板——二维树状数组之求和(区间修改,区间查询)

#include<iostream>
using namespace std;

typedef long long ll;
const int N = 2100;


ll n, m;
ll t1[N][N], t2[N][N];
ll t4[N][N], t3[N][N];

ll lowbit(ll x) { return x & (-x); }
void Update(ll x, ll y, ll k) {
	for (ll i = x; i <= n; i += lowbit(i)) {
		for (ll j = y; j <= m; j += lowbit(j)) {
			t1[i][j] += k;
			t2[i][j] += x * k;
			t3[i][j] += y * k;
			t4[i][j] += x * y * k;
		}
	}
}
ll sum(ll x, ll y) {
	ll ans = 0;
	for (ll i = x; i >= 1; i -= lowbit(i)) {
		for (ll j = y; j >= 1; j -= lowbit(j)) {
			ans += (x + 1) * (y + 1) * t1[i][j] - (y + 1) * t2[i][j] - (x + 1) * t3[i][j] + t4[i][j];
		}
	}
	return ans;
}
int main() {

	cin >> n >> m;

	int sign;
	while (cin >> sign) {
		if (sign == 1) {		//区间修改
			ll x1, y1, x2, y2, k;
			cin >> x1 >> y1 >> x2 >> y2 >> k;
			Update(x1, y1, k);		//二维差分思想
			Update(x1, y2 + 1, -k);
			Update(x2 + 1, y1, -k);
			Update(x2 + 1, y2 + 1, k);
		}
		else {		//区间查询
			ll x1, y1, x2, y2;
			cin >> x1 >> y1 >> x2 >> y2;
			ll ans = sum(x2, y2) - sum(x1 - 1, y2) - sum(x2, y1 - 1) + sum(x1 - 1, y1 - 1);
			cout << ans << endl;
		}
	}
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值