BZOJ 4551: [Tjoi2016&Heoi2016]树 并查集

这年的HEOI 是在数据有点水，据说一堆写BFS都100了。正解使用并查集反向维护，感觉是不是很难，毕竟DAY1 T1。

树链剖分接线段树是可做的，但是吧…代码量瞬间就上来了。

主体思想，把正向加点改为反向删点。 每个并查集的代表元素为被标记的点 ，集合存的是答案。删点时直接将并查集合并即可。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int maxn = 5e5+500;
struct ask
{
	int tp,x;	
};
ask T[maxn];
vector<int> f[maxn];
char op[5];
int n,q,x,y;
int Fa[maxn],ans[maxn];
int vis[maxn];
int F[maxn];
int find(int x)
{
	return F[x]==x ? x : F[x]=find(F[x]) ;
}
void dfs(int u,int fa)
{
	Fa[u]=fa;
	for(int i=0;i<f[u].size();i++)
	{
		int v=f[u][i];
		if(v==fa) continue;
		dfs(v,u);
	}
}
int main()
{
	scanf("%d%d",&n,&q);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		f[x].push_back(y);
		f[y].push_back(x);
	}
	vis[1]=1;
	for(int i=1;i<=q;i++)
	{
		scanf("%s%d",op,&x);
		if(op[0]=='Q'){
			T[i].tp=1;T[i].x=x;
		}else{
			T[i].tp=2;T[i].x=x;
			vis[x]++;
		}
	}
	dfs(1,0);
	for(int i=1;i<=n;i++) F[i]= vis[i] ? i : Fa[i];
	for(int i=q;i>=1;i--)
	{
		if(T[i].tp==2){
			vis[T[i].x]--;
			if(!vis[T[i].x])	F[T[i].x]=Fa[T[i].x];
		}else{
			ans[i]=find(T[i].x);
		}
	}
	for(int i=1;i<=q;i++)
	{
		if(T[i].tp==1) cout<<ans[i]<<endl;
	} 
	return 0;
}