SP7826 TREEISO - Tree Isomorphism

题意

判断树同构。

分析

直接树哈希即可。

代码

#include<bits/stdc++.h>
using namespace std;
template <typename T>
inline void read(T &x){
	x=0;char ch=getchar();bool f=false;
	while(!isdigit(ch)){if(ch=='-'){f=true;}ch=getchar();}
	while(isdigit(ch)){x=x*10+(ch^48);ch=getchar();}
	x=f?-x:x;
	return ;
}
template <typename T>
inline void write(T x){
	if(x<0) putchar('-'),x=-x;
	if(x>9) write(x/10);
	putchar(x%10^48);
	return ;
}
#define ll long long
#define PII pair<int,int>
#define mp make_pair
#define pb push_back
#define fi first
#define se second
const int N=1e5+5,M=2e6+5;
int n,m,p,prime[M],cnt,siz[N],f[N],g[N],rt;
vector<int>vec[N];
bool vis[M];
void Primes(int n){
	for(int i=2;i<=n;i++){
		if(!vis[i]) prime[++cnt]=i;
		for(int j=1;i*prime[j]<=n&&j<=cnt;j++){
			vis[i*prime[j]]=true;
			if(i%prime[j]==0) break;
		}
	}
	return ;
}
void dfs1(int x,int fa){
	f[x]=1;siz[x]=1;
	for(auto y:vec[x]){if(y==fa) continue;dfs1(y,x);siz[x]+=siz[y];f[x]+=f[y]*prime[siz[y]];}
	return ;
}
void dfs2(int x,int fa,int val){
	val=val*prime[n-siz[x]];
	g[x]=f[x]+val;
	for(auto y:vec[x]) if(y!=fa) dfs2(y,x,val+f[x]-f[y]*prime[siz[y]]);
	return ;
}
int now=n,rt1=1e5+1,rt2=1e5+1;
void GetRoot(int x,int fa){
	int Maxn=0;
	for(auto y:vec[x]){
		if(y==fa) continue;
		GetRoot(y,x);
		Maxn=max(Maxn,siz[y]);
	}
	Maxn=max(Maxn,n-siz[x]);
	if(Maxn<now) now=Maxn,rt1=x,rt2=1e5+1;
	else if(Maxn==now) rt2=x;
	return ;
}
PII v[N];
int main(){
	read(m);Primes(2e6);
	for(int t=1;t<=m;t++){
		read(n);
		for(int i=1;i<=n;i++) vec[i].clear();
		for(int i=1;i<n;i++){
			int u,v;
			read(u),read(v);
			vec[u].pb(v),vec[v].pb(u);
		}
		rt=1;
		dfs1(rt,0);dfs2(rt,0,0);
		rt1=rt2=1e5+1,now=n;
		GetRoot(rt,0);
		v[1]=mp(g[rt1],g[rt2]);
		bool flag=false;
		for(int i=1;i<=n;i++) vec[i].clear();
		for(int i=1;i<n;i++){
			int u,v;
			read(u),read(v);
			vec[u].pb(v),vec[v].pb(u);
		}
		rt=1;
		dfs1(rt,0);dfs2(rt,0,0);
		rt1=rt2=1e5+1,now=n;
		GetRoot(rt,0);
		v[2]=mp(g[rt1],g[rt2]);
		if(v[1].se&&v[2].se){if(v[1].fi==v[2].fi||v[1].fi==v[2].se||v[1].se==v[2].fi||v[1].se==v[2].se){flag=true;}}
		else if(v[1].se){if(v[1].fi==v[2].fi||v[1].se==v[2].fi){flag=true;}}
		else if(v[2].se){if(v[1].fi==v[2].fi||v[1].fi==v[2].se){flag=true;}}
		else{if(v[1].fi==v[2].fi){flag=true;}}
		puts(flag?"YES":"NO");
	}
	return 0;
}