CodeForces - 707E Garlands (二维树状数组)

傻逼题，能不能AC取决于眼力够不够好能看到“询问次数不超过2000”这一限制

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
const int N=2000+10,inf=0x3f3f3f3f;
int n,m,k,Q,nqr,state[N];
char buf[N];
ll c[N][N],ans[N];
int lb(int x) {return x&-x;}
void add(int u,int v,int x) {
    for(int i=u; i<=n; i+=lb(i))
        for(int j=v; j<=m; j+=lb(j))c[i][j]+=x;
}
ll get(int u,int v) {
    ll ret=0;
    for(int i=u; i; i-=lb(i))
        for(int j=v; j; j-=lb(j))ret+=c[i][j];
    return ret;
}
struct P {int x,y,c;};
vector<P> vec[N];
struct Q {int x1,y1,x2,y2;} qr[N];
vector<int> gx[N];

int main() {
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1; i<=k; ++i) {
        int l;
        scanf("%d",&l);
        while(l--) {
            int x,y,c;
            scanf("%d%d%d",&x,&y,&c);
            vec[i].push_back({x,y,c});
        }
    }
    for(int i=1; i<=k; ++i)state[i]=1;
    for(scanf("%d",&Q); Q--;) {
        int x1,x2,y1,y2,u;
        scanf("%s",buf);
        if(buf[0]=='S')scanf("%d",&u),state[u]^=1;
        else {
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            qr[++nqr]= {x1,y1,x2,y2};
            for(int i=1; i<=k; ++i)if(state[i])gx[i].push_back(nqr);
        }
    }
    for(int i=1; i<=k; ++i) {
        for(P p:vec[i])add(p.x,p.y,p.c);
        for(int x:gx[i]) {
            int x1=qr[x].x1,y1=qr[x].y1,x2=qr[x].x2,y2=qr[x].y2;
            ans[x]+=get(x2,y2)-get(x1-1,y2)-get(x2,y1-1)+get(x1-1,y1-1);
        }
        for(P p:vec[i])add(p.x,p.y,-p.c);
    }
    for(int i=1; i<=nqr; ++i)printf("%lld\n",ans[i]);
    return 0;
}