题意:有q次操作,每次操作 (x ,y),如果S集合有(x ,y),就删除这对数,如果没有就插入这对数,然后询问S集合在R规则下一共有多少对数(R规则:如果R集合中有(x1,y1) (x1,y2)(x2,y1),且没有(x2,y2),那么在计数的时候会加上(x2,y2)这对)。
思路:第一次写线段树分治+可撤销并查集,代码借鉴:幽蝶,如果一对数出现奇数次,那么是有贡献,偶数次无贡献,贡献就是每个联通块中的x点集的大小*y点集的大小,我们先离线处理每队数贡献区间(即在 l 操作到 r 操作这段区间出现次数为奇数次),更新到线段树区间上,然后遍历整个线段树求总贡献,求到叶子节点[ x,x ]就是第x次操作的答案。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=3e5+10,N=3e5;
struct node
{
int x,y;
node(){;}
node(int a,int b){x=a,y=b;}
bool operator==(const node&t)const
{
return x==t.x&&y==t.y;
}
bool operator<(const node&t)const
{
if(x==t.x)return y>t.y;
return x<t.x;
}
}a[maxn],b[maxn];
vector<node>tree[maxn*4];
stack<node>s;
int p[maxn*2],vis[maxn],cnt,sz[maxn*2],sz1[maxn],sz2[maxn*2];
ll ans[maxn];
int find(int x)
{
if(x!=p[x])
return find(p[x]);
return p[x];
}
void up(int o,int l,int r,int ql,int qr,int x,int y)
{
if(l>=ql&&r<=qr)
{
tree[o].push_back(node(x,y));
return;
}
int m=(l+r)/2,ls=o*2,rs=o*2+1;
if(ql<=m)up(ls,l,m,ql,qr,x,y);
if(qr>m)up(rs,m+1,r,ql,qr,x,y);
}
void dfs(int o,int l,int r,ll res)
{
int top=s.size(),m=(l+r)/2;
for(int i=0;i<tree[o].size();i++)
{
int u=find(tree[o][i].x);
int v=find(tree[o][i].y);
if(u==v)continue;
if(sz[u]<sz[v])swap(u,v);
sz[u]+=sz[v];
p[v]=u;
s.push(node(u,v));
res-=1ll*sz1[u]*sz2[u]+1ll*sz1[v]*sz2[v];
sz1[u]+=sz1[v];sz2[u]+=sz2[v];
res+=1ll*sz1[u]*sz2[u];
}
if(l==r)
ans[l]=res;
else
{
dfs(o*2,l,m,res);
dfs(o*2+1,m+1,r,res);
}
while(s.size()>top)
{
int u=s.top().x,v=s.top().y;
s.pop();
sz[u]-=sz[v];
p[v]=v;
sz1[u]-=sz1[v],sz2[u]-=sz2[v];
}
}
int main()
{
int q;
scanf("%d",&q);
for(int i=1;i<=q;i++)
{
scanf("%d%d",&a[i].x,&a[i].y);a[i].y+=N;
b[i]=a[i];
}
sort(b+1,b+1+q);
int n=unique(b+1,b+1+q)-b-1;
for(int i=1;i<=q;i++)
{
int cur=lower_bound(b+1,b+1+n,a[i])-b;
if(!vis[cur])vis[cur]=i;
else
{
up(1,1,q,vis[cur],i-1,a[i].x,a[i].y);
vis[cur]=0;
}
}
for(int i=1;i<=n;i++)
if(vis[i])
up(1,1,q,vis[i],q,b[i].x,b[i].y);
for(int i=1;i<=N;i++)
{
p[i]=i;p[i+N]=i+N;
sz1[i]=sz2[i+N]=sz[i]=sz[i+N]=1;
}
dfs(1,1,q,0);
for(int i=1;i<=q;i++)
printf("%lld ",ans[i]);
}