[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;
}
posted @ 2019-07-07 12:58  QwQGJH  阅读(131)  评论(0编辑  收藏  举报