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

来源:互联网 发布:mac怎么去掉dashboard 编辑:程序博客网 时间:2024/06/16 12:36

题面
树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;}
原创粉丝点击