题意: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;
}