Codeforces Round #368 (Div. 2) E. Garlands(查询矩形内的和)

传送门:Codeforces Round #368 (Div. 2)

---

题意:2000*2000的棋盘 有k个不相交集合,每个集合点数<=2000,然后10^6次操作,要么是一个集合从出现变成消失,消失变成出现
要么查询一个矩形内点数,查询数不超过2000,集合数<=2000,q<=1e6

---

思路:因为集合的个数比较少,所以我们可以暴力找到哪些集合是开的,然后对于每个集合查询(x1,y1)-(x2,y2)中的价值总和

方案一:直接暴力开k棵主席树
方案二:按k去查询

---

树状数组维护:

#include<bits/stdc++.h>
using namespace std;
const int maxn=2001;
struct Point{
    int x,y,w;
    bool operator <(const Point &r) const{
        return r.x==x ?  y<r.y : x<r.x;
    }
}a[maxn][maxn];
struct Node{
    int id,x1,y1,x2,y2;
};
vector<Node>tmp[maxn];
long long ans[maxn],C[maxn][maxn];
int Belong[maxn][maxn],len[maxn];
int flag[maxn],n,m;
char s[10];

void add(int x,int y,int id,int w){
    for(int i=x;i<=n;i+=(i&-x))
        for(int j=y;j<=m;j+=(j&-j)){
            if(Belong[i][j]==id)
                C[i][j]+=w;
            else
                C[i][j]=w,Belong[i][j]=id;
        }
}

long long sum(int x,int y,int id){
    long long ret=0;
    for(int i=x;i>0;i-=(i&-i))
        for(int j=y;j>0;j-=(j&-j))
            if(Belong[i][j]==id)
                ret+=C[i][j];
    return ret;
}

int main(){
    int q,k,x,x1,y1,x2,y2,cnt=0;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=k;i++)
        flag[i]=1;
    for(int i=1;i<=k;i++){
        scanf("%d",&len[i]);
        for(int j=1;j<=len[i];j++)
            scanf("%d%d%d",&a[i][j].x,&a[i][j].y,&a[i][j].w);
        sort(a[i]+1,a[i]+len[i]+1);
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        scanf("%s",s);
        if(s[0]=='S')
            scanf("%d",&x),flag[x]^=1;
        else{
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            ++cnt;
            for(int j=1;j<=k;j++)
                if(flag[j]==1)
                    tmp[j].push_back( (Node) {cnt,x1,y1,x2,y2} );
        }
    }
    for(int i=1;i<=k;i++){
        for(int j=1;j<=len[i];j++)
            add(a[i][j].x,a[i][j].y,i,a[i][j].w);
        for(int j=0;j<tmp[i].size();j++)
            ans[tmp[i][j].id]+=sum(tmp[i][j].x2,tmp[i][j].y2,i)-sum(tmp[i][j].x1-1,tmp[i][j].y2,i)-sum(tmp[i][j].x2,tmp[i][j].y1-1,i)+sum(tmp[i][j].x1-1,tmp[i][j].y1-1,i);
    }
    for(int i=1;i<=cnt;i++)
        printf("%lld\n",ans[i]);
    return 0;
}

主席树:

#include<bits/stdc++.h>
using namespace std;
const int maxn=2001;
struct Point{
    int x,y,w;
    bool operator <(const Point &r) const{
        return r.x==x ?  y<r.y : x<r.x;
    }
}a[maxn][maxn];
struct Node{
    int id,x1,y1,x2,y2;
};
vector<Node>tmp[maxn];

int root[maxn],sz,b[maxn],len[maxn];
long long ans[maxn];

struct node{
    int l,r;
    long long w;
}T[maxn*20];

void update(int &i,int l,int r,int num,int w){
    T[++sz]=T[i],i=sz;
    T[i].w+=w;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    if(num<=mid)
        update(T[i].l,l,mid,num,w);
    else
        update(T[i].r,mid+1,r,num,w);
}

long long query(int x,int y,int l,int r,int L,int R){
    if(L<=l&&R>=r)
        return T[y].w-T[x].w;
    int mid=(l+r)>>1;
    long long Ans=0;
    if(L<=mid)
        Ans+=query(T[x].l,T[y].l,l,mid,L,R);
    if(R>mid)
        Ans+=query(T[x].r,T[y].r,mid+1,r,L,R);
    return Ans;
}

int flag[maxn];
char s[10];

int main(){
    int n,m,q,k,x,x1,y1,x2,y2,cnt=0;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=k;i++)
        flag[i]=1;
    for(int i=1;i<=k;i++){
        scanf("%d",&len[i]);
        for(int j=1;j<=len[i];j++)
            scanf("%d%d%d",&a[i][j].x,&a[i][j].y,&a[i][j].w);
        sort(a[i]+1,a[i]+len[i]+1);
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        scanf("%s",s);
        if(s[0]=='S')
            scanf("%d",&x),flag[x]^=1;
        else{
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            ++cnt;
            for(int j=1;j<=k;j++)
                if(flag[j]==1)
                    tmp[j].push_back( (Node) {cnt,x1,y1,x2,y2} );
        }
    }
    for(int i=1;i<=k;i++){
        sz=0;
        int z=1;
        for(int j=1;j<=n;j++){
            root[j]=root[j-1];
            while(z<=len[i]&&a[i][z].x==j)
                update(root[j],1,m,a[i][z].y,a[i][z].w),z++;
        }
        for(int j=0;j<tmp[i].size();j++)
            ans[tmp[i][j].id]+=query(root[ tmp[i][j].x1-1 ],root[ tmp[i][j].x2 ],1,m,tmp[i][j].y1,tmp[i][j].y2);
    }
    for(int i=1;i<=cnt;i++)
        printf("%lld\n",ans[i]);
    return 0;
}