[2017纪中11-5]仔细的检查 树hash

题面
树hash裸题。
正确姿势貌似是找重心，然后每次计算x点的hash值要把其子树的hash值排序之后再像字符串那样hash，最后融进自己的一些信息去。
我则是乱搞，每个点的hash值是其儿子的hash值的和+R（size[x]，num[x]）（关于其子树大小和儿子个数的一个函数，千万不要是一次的。。。）这样就可以O(1)换根啦。
找到一个相同的根，两个再一起跑一边DFS就可以啦。
代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ull unsigned long long
#define R(x,y) ((ull)((x)*(x)+(y)*(y)+233)*((x)+(y)+666)) 
using namespace std;
const int maxn=100010;
int n,sz[2][maxn],num[2][maxn],rt,ans[maxn];
ull h[2][maxn];
struct node
{
    ull t;int id;
}z0[maxn],z1[maxn];
bool cmp(node a,node b){return a.t<b.t;}
struct edge
{
    int t;
    edge *next;
}*con[2][maxn];
void ins(int b,int x,int y)
{
    edge *p=new edge;
    p->t=y;
    p->next=con[b][x];
    con[b][x]=p;
}
void dfs(int b,int v,int fa)
{
    sz[b][v]=1;h[b][v]=0;num[b][v]=0;
    for(edge *p=con[b][v];p;p=p->next)
        if(p->t!=fa)
        {
            dfs(b,p->t,v);
            sz[b][v]+=sz[b][p->t];
            h[b][v]+=h[b][p->t];
            num[b][v]++;
        }
    h[b][v]+=R(sz[b][v],num[b][v]); 
}
void frt(int v,int fa)
{
    if(fa!=0)
    {
        ull hf=h[1][fa]-R(n,num[1][fa]+(fa!=1))-h[1][v]+R(n-sz[1][v],num[1][fa]-(fa==1));
        h[1][v]=h[1][v]-R(sz[1][v],num[1][v])+R(n,num[1][v]+1)+hf;
    }
    if(h[1][v]==h[0][1]) {rt=v;return;}
    for(edge *p=con[1][v];p;p=p->next)
        if(p->t!=fa)
        {
            frt(p->t,v);
            if(rt!=0) return ;
        }
}
void qry(int v0,int v1,int fa0,int fa1)
{
    int top=0;
    for(edge *p=con[0][v0];p;p=p->next)
        if(p->t!=fa0) z0[++top].t=h[0][p->t],z0[top].id=p->t;
    top=0;  
    for(edge *p=con[1][v1];p;p=p->next)
        if(p->t!=fa1) z1[++top].t=h[1][p->t],z1[top].id=p->t;   
    sort(z0+1,z0+top+1,cmp);
    sort(z1+1,z1+top+1,cmp);
    for(int i=1;i<=top;i++)
        ans[z0[i].id]=z1[i].id;
    for(edge *p=con[0][v0];p;p=p->next)
        if(p->t!=fa0) qry(p->t,ans[p->t],v0,v1);        
}
int main()
{
    scanf("%d",&n);
    int x,y;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        ins(0,x,y);
        ins(0,y,x);
    }
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        ins(1,x,y);
        ins(1,y,x);
    }

    dfs(0,1,0);dfs(1,1,0);
    rt=0;frt(1,0);
    if(rt>0) puts("YES");
    else {puts("NO");return 0;}
    ans[1]=rt;
    dfs(1,rt,0);
    qry(1,rt,0,0);

    for(int i=1;i<=n;i++)
        printf("%d ",ans[i]);
    return 0;
}