Codeforces Round #368 (Div. 2) E. Garlands 二维树状数组
题目链接
Like all children, Alesha loves New Year celebration. During the celebration he and his whole family dress up the fir-tree. Like all children, Alesha likes to play with garlands — chains consisting of a lightbulbs.
Alesha uses a grid field sized n × m for playing. The rows of the field are numbered from 1 to n from the top to the bottom and columns are numbered from 1 to m from the left to the right.
Alesha has k garlands which he places at the field. He does so in the way such that each lightbulb of each garland lies in the center of some cell in the field, and each cell contains at most one lightbulb. Of course lightbulbs, which are neighbours in some garland, appears in cells neighbouring by a side.
Each garland is turned off or turned on at any moment. If some garland is turned on then each of its lightbulbs is turned on, the same applies for garland turned off. Each lightbulb in the whole garland set is unique, and thus, being turned on, brings Alesha some pleasure, described by an integer value. Turned off lightbulbs don't bring Alesha any pleasure.
Alesha can turn garlands on and off and wants to know the sum of pleasure value which the lightbulbs, placed in the centers of the cells in some rectangular part of the field, bring him. Initially all the garlands are turned on.
Alesha is still very little and can't add big numbers. He extremely asks you to help him.
InputThe first line of the input contains three integers n, m and k (1 ≤ n, m, k ≤ 2000) — the number of field rows, the number of field columns and the number of garlands placed at the field respectively.
Next lines contains garlands set description in the following format:
The first line of a single garland description contains a single integer len (1 ≤ len ≤ 2000) — the number of lightbulbs in the garland.
Each of the next len lines contains three integers i, j and w (1 ≤ i ≤ n, 1 ≤ j ≤ m, 1 ≤ w ≤ 109) — the coordinates of the cell containing a lightbullb and pleasure value Alesha gets from it if it is turned on. The lightbulbs are given in the order they are forming a chain in the garland. It is guaranteed that neighbouring lightbulbs are placed in the cells neighbouring by a side.
The next line contains single integer q (1 ≤ q ≤ 106) — the number of events in Alesha's game. The next q lines describes events in chronological order. The i-th of them describes the i-th event in the one of the following formats:
- SWITCH i — Alesha turns off i-th garland if it is turned on, or turns it on if it is turned off. It is guaranteed that 1 ≤ i ≤ k.
- ASK x1 y1 x2 y2 — Alesha wants to know the sum of pleasure values the lightbulbs, placed in a rectangular part of the field. Top-left cell of a part has coordinates (x1, y1) and right-bottom cell has coordinates (x2, y2). It is guaranteed that 1 ≤ x1 ≤ x2 ≤ n and 1 ≤ y1 ≤ y2 ≤ m. There is no more than 2000 events of this type in the input.
All the numbers in the input are integers.
Please note that the input is quite large, so be careful while using some input ways. In particular, it's not recommended to use cin in codes on C++ and class Scanner in codes on Java.
For each ASK operation print the sum Alesha wants to know in a separate line. Print the answers in chronological order.
4 4 3 5 1 1 2 1 2 3 2 2 1 2 1 4 3 1 7 4 1 3 1 2 3 3 2 4 3 1 4 1 7 4 1 1 4 2 9 3 2 8 3 3 3 4 3 4 4 4 1 3 4 1 2 ASK 2 2 3 3 ASK 1 1 4 4
15 52
4 4 1 8 4 1 1 3 1 2 2 1 1 1 1 7 1 2 5 2 2 4 2 3 1 1 3 1 3 ASK 1 1 3 2 SWITCH 1 ASK 1 1 3 2
19 0
题意：n*m的矩形空地，有k条串联的灯，每个灯只能占一格，相邻的灯在相邻的空地上。有q个操作，每次可以开关一个链上的灯，或查询一个矩形的权值和。
思路：当要频繁的对数组元素进行修改，同时又要频繁的查询数组内任一区间元素之和的时候，可以考虑使用树状数组。 这儿用二维树状数组。修改操作可以先记着，等到查询的时候再更新树状数组。
#include
using namespace std;
const int maxn = 2e3+5;
int n, m;
long long C[maxn][maxn];
int lowbit(int x)
{
return x&(-x);
}
int Add(int x, int y, int d){
for(int i=x; i<=n; i +=lowbit(i))
for(int j=y; j<=m; j+=lowbit(j))
C[i][j] += d;
}
long long Sum(int x, int y){
long long res = 0;
for(int i=x; i>0; i-=lowbit(i))
for(int j=y; j>0; j-=lowbit(j))
res += C[i][j];
return res;
}
vector vex[maxn],vey[maxn],vew[maxn];
int main()
{
int k, num, x, y, z, q;
char op[10];
int flag[maxn], light[maxn];
memset(flag,0,sizeof(flag));
memset(light,0,sizeof(light));
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=k;i++){
scanf("%d",&num);
light[i]=1;
for(int j=1;j<=num;j++){
scanf("%d%d%d",&x,&y,&z);
vex[i].push_back(x);
vey[i].push_back(y);
vew[i].push_back(z);
}
}
scanf("%d",&q);
for(int i=1;i<=q;i++){
scanf("%s",op);
if(op[0]=='S'){
int a;
scanf("%d",&a);
flag[a] ^= 1;
}
else{
int a, b,c,d;
scanf("%d%d%d%d",&a,&b,&c,&d);
for(int j=1;j<=k;j++){
if(light[j]==flag[j]) continue;
if(flag[j]==0){
for(int ii=0;ii