[JSOI2016]独特的树叶

Posted 2022-03-23 mle-world

仙题+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 nodeint 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;