P4323 [JSOI2016]独特的树叶

题目

分析

显然如果我们某个时候把这个多余的叶子指定为根的话，那么其子节点的树哈希值就应该等于原树某个点的哈希值。

于是可以考虑直接换根dp，第一个树直接把每一个节点的值都扔进一个哈希表，第二个树的时候可以对于每一个节点作为根的时候，消去自己的贡献然后判断是否存在这个消去后的数即可。

代码

#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
#define ull unsigned long long
const int N=1e5+5,M=2e6+5,INF=1e9+7;
unordered_map<ull,int>Map;
int n,m,p,cnt,siz[N],rt,Ans=INF;
vector<int>vec[N];
bool vis[M];
ull prime[M],f[N],g[N];
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){dfs1(y,x);siz[x]+=siz[y];f[x]+=f[y]*prime[siz[y]];}
	return ;
}
void dfs2(int x,int fa,ull val,bool fl){
	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]],fl);
	if(fl){Map[g[x]]=1;}
	else if(vec[x].size()==1){if(Map[g[vec[x][0]]-prime[1]]) Ans=min(Ans,x);}
	return ;
}
int fa[N],ad[N],v[N];
bool vv[N];
int Getfa(int x){return x==fa[x]?x:fa[x]=Getfa(fa[x]);}
void Merge(int x,int y){
	x=Getfa(x),y=Getfa(y);
	if(x==y) return ;
	fa[x]=y;
	return ;
}
int main(){
	read(n);Primes(2e6);
	for(int i=1;i<=n;i++) vec[i].clear();
	for(int i=1,x,y;i<n;i++){read(x),read(y),vec[x].pb(y),vec[y].pb(x);}
	rt=1;
	dfs1(rt,0);dfs2(rt,0,0,1);
	n++;
	for(int i=1;i<=n;i++) vec[i].clear();
	for(int i=1,x,y;i<n;i++){read(x),read(y),vec[x].pb(y),vec[y].pb(x);}
	rt=1;
	dfs1(rt,0);dfs2(rt,0,0,0);
	write(Ans);
	return 0;
}