[JSOI2016]独特的树叶
仙题+1.
我们发现如果能够求出来\(A\)树中任何一个点当根的时候的\(Hash\)值,那么就可以求出答案了。
然后你随便写一写\(Hash\)策略改成\(xor\),发现\(xor\)的逆运算就是\(xor\),好啊!
换根\(dp\)直接求出我们需要的,然后放到\(set\)去就行了。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<iostream>
using namespace std;
#define re register
#define ll long long
inline int gi(){
int f=1,sum=0;char ch=getchar();
while(ch>'9' || ch<'0'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}
return f*sum;
}
const int N=100010;
typedef unsigned int ull;
const ull base=19491001;
map<ull,bool>mp;
struct Tree{
int front[N],cnt,n,siz[N];
struct node{int to,nxt;}e[N<<1];
void Add(int u,int v){e[++cnt]=(node){v,front[u]};front[u]=cnt;}
ull f[N],g[N];
void dfs1(int u,int ff){
siz[u]=1;
for(int i=front[u];i;i=e[i].nxt){
int v=e[i].to;if(v==ff)continue;
dfs1(v,u);
f[u]=f[u]^(f[v]*base+siz[v]);siz[u]+=siz[v];
}
}
void dfs2(int u,int ff){
if(ff)g[u]=f[u]^((g[ff]^(f[u]*base+siz[u]))*base+n-siz[u]);
else g[u]=f[u];
for(int i=front[u];i;i=e[i].nxt){
int v=e[i].to;if(v==ff)continue;
dfs2(v,u);
}
}
}A,B;
int d[N];
int main(){
int n=gi();A.n=n;B.n=n+1;
for(int i=1;i<n;i++){int u=gi(),v=gi();A.Add(u,v);A.Add(v,u);}
for(int i=1;i<=n;i++){int u=gi(),v=gi();d[u]++;d[v]++;B.Add(u,v);B.Add(v,u);}
A.dfs1(1,0);A.dfs2(1,0);
for(int i=1;i<=n;i++)mp[A.g[i]]=1;
for(int i=1;i<=n+1;i++)
if(d[i]>1){B.dfs1(i,0);B.dfs2(i,0);break;}
for(int i=1;i<=n+1;i++)
if(d[i]==1){
int v=B.e[B.front[i]].to;
ull w=B.g[v]^(B.f[i]*base+1);
if(mp[w]){
printf("%d\n",i);
return 0;
}
}
return 0;
}