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;
}