Educational Codeforces Round 62 F. Extending Set of Points 线段树分治+可撤销并查集

F. Extending Set of Points

题意：有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]);
}