码迷,mamicode.com
首页 > Web开发 > 详细

[JSOI2016]独特的树叶

时间:2019-07-07 14:49:38      阅读:137      评论:0      收藏:0      [点我收藏+]

标签:gis   define   reg   include   signed   bre   using   sig   pac   

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

[JSOI2016]独特的树叶

标签:gis   define   reg   include   signed   bre   using   sig   pac   

原文地址:https://www.cnblogs.com/mle-world/p/11146013.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!