Codeforces 390E Inna and Large Sweet Matrix(树状数组）

题目链接：Codeforces 390E Inna and Large Sweet Matrix

题目大意：给出n，m和w，表示在一个n*m的图上进行操作，0表示添加操作，在点（x1，y1)和(x2,y2)之间的所有格子上添加value个糖果；1表示查询操作，查询（x1，y1)和(x2,y2)之间糖果的数量。

解题思路：树状数组，将x，y分开讨论，bit1进行正常的添加操作，bit2则进行周边处理。

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>

using namespace std;
typedef long long ll;
const int N = 4 * 1000005;

int n[2], w;
ll bit1[2][N], bit2[2][N];

void insert(ll* b, int m, int x, ll value) {
	while (x <= m) {
		b[x] += value;
		x += x & (-x);
	}
}

ll get(ll* b, int x) {
	ll ans = 0;
	while (x > 0) {
		ans += b[x];
		x -= x & (-x);
	}
	return ans;
}

ll Get(int id, int r) {
	return get(bit1[id], r) * r + get(bit2[id], r);
}

void Insert(int id, int l, int r, ll value) {
	insert(bit1[id], n[id], l, value);
	insert(bit1[id], n[id], r+1, -value);
	insert(bit2[id], n[id], l, -value * (l-1));
	insert(bit2[id], n[id], r+1, value * r);
}

int main () {
	int x1, x2, y1, y2, v;
	scanf("%d%d%d", &n[0], &n[1], &w);
	while (w--) {
		scanf("%d%d%d%d%d", &v, &x1, &y1, &x2, &y2);
		if (v) {
			ll ans = Get(1, y2) - Get(1, y1-1) - (Get(0, n[0]) - Get(0, x2) + Get(0, x1-1));
			cout << ans << endl;
		} else {
			ll value;
			cin >> value;
			Insert(0, x1, x2, value * (y2-y1+1));
			Insert(1, y1, y2, value * (x2-x1+1));
		}
	}
	return 0;
}