codeforces 690 F3 Tree of Life (hard) 树hash

最新推荐文章于 2022-01-19 23:17:22 发布

make_it_for_good

最新推荐文章于 2022-01-19 23:17:22 发布

阅读量562

点赞数 1

分类专栏：树hash 文章标签： codeforces hash

本文链接：https://blog.csdn.net/make_it_for_good/article/details/52452650

版权

树hash 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

题意：一棵树，给出两个分别去掉两个不同点之后形成的图，求这棵树原来长什么样。 $n<=1000$

我从未见过如此恶心的树hash

树大概长这样，u，v分别是去掉的两个点。

这里写图片描述

两个图分别长这样：
这里写图片描述

和这样：
这里写图片描述

把u，v都去掉就变成这样：
这里写图片描述

可以看到这个图被分成了3部分：
连接u，v的部分
包含u的子树去掉u，再去掉u，v之间部分剩下的一堆子树
包含v的子树去掉v，再去掉u，v之间部分剩下的一堆子树

有一个特殊情况就是第一部分没有点，就是一条边，特判一下就好了。

好，那我们只需要从第一个图里找一个v，从第二个图里找一个u，
如果第一个图的三个部分和第二个图的三个部分完全相同的话，我们就找到了一组解。

如何判断两个图的三个部分是否对应相同？
分别求出三个部分的hash值，再把三个部分hash到一起。
对于一堆子树就求出每个子树的hash值再hash到一起。
注意这两个hash和树hash需要三个不同的seed。

那我对于两个图分别搞，对于每个图求出所有满足条件的三个部分的hash值，然后把两个图的hash值分别排个序，找相等的hash值就行了。
一个图有 $O(n)$ 种三个部分的取法，这个一会说。

这个东西扫一遍一个图的hash值，一个指针找另一个图的hash值可以做到 $O(nlogn)$ 。

对于一个图，枚举所有子树，让枚举到的子树成为两个部分，其他的子树成为一个部分。

枚举子树中的点，求出这个子树中这个点的所有子树的hash值。这步 $O(n^2)$
再枚举u，v之间的子树是谁（这个只需要枚举当前点连出的一条边就行了，因此一共有 $O(n)$ 种取法）。
然后求一下hash值就好了。

总复杂度 $O(n^2)$

写了一下午，顺便一提，我的还是最短代码。

#include <bits/stdc++.h>
using namespace std;
#define N 1100
#define seed 11333333
#define seed2 5462617
#define seed1 23333333
#define ull unsigned long long
int n,T,label,pos[2][N];
struct node
{
    ull v;int x,y;
    node(){}
    node(ull v,int x,int y):v(v),x(x),y(y){}
    friend bool operator < (const node &r1,const node &r2)
    {return r1.v<r2.v;};
};
vector<node>a[2];
struct tree
{
    int id,m,f1,f2,sum;
    int used[N],f[N],size[N],vis[N];
    vector<int>  v[N],sub[N];
    vector<ull>  v1[N],v2;
    vector<node> v3;
    ull val[N];
    void init()
    {
        a[id].clear();
        memset(used,0,sizeof(used));
        memset(vis,0,sizeof(vis));
        v2.clear();
        for(int i=1;i<=n;i++)
            v[i].clear(),sub[i].clear();
    }
    void getsub(int x,int y,int root)
    {
        sub[root].push_back(x);
        for(int i=0,t;i<v[x].size();i++)
            if((t=v[x][i])!=y)
                getsub(t,x,root);
    }
    void dfs1(int x,int y)
    {
        size[x]=1;used[x]=1;
        for(int i=0,t;i<v[x].size();i++)
            if((t=v[x][i])!=y&&!vis[t])
                dfs1(t,x),size[x]+=size[t];
    }
    void dfs2(int x,int y)
    {
        f[x]=sum-size[x];
        for(int i=0,t;i<v[x].size();i++)
            if((t=v[x][i])!=y&&!vis[t])
                dfs2(t,x),f[x]=max(f[x],size[t]);
        if(f[x]<f[f1])f1=x;
        else if(f[x]<f[f2])f2=x;    
    }
    ull dfs3(int x,int y)
    {
        v1[x].clear();
        for(int i=0,t;i<v[x].size();i++)
            if((t=v[x][i])!=y&&!vis[t])
                v1[x].push_back(dfs3(t,x));
        sort(v1[x].begin(),v1[x].end());
        ull ret=1;
        for(int i=0;i<v1[x].size();i++)
            ret=ret*seed+v1[x][i];
        return ret;
    }
    ull cal(int x)
    {
        f1=f2=0;f[0]=1<<30;
        dfs1(x,0);
        sum=size[x];dfs2(x,0);
        if(f[f1]!=f[f2])
            return dfs3(f1,0);
        ull t1=dfs3(f1,f2);
        ull t2=dfs3(f2,f1);
        if(t1>t2)swap(t1,t2);
        return t1*seed+t2;
    }
    void solve(int x)
    {
        id=x;init();
        scanf("%d",&m);
        for(int i=1,x,y;i<=m;i++)
        {
            scanf("%d%d",&x,&y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        for(int i=1;i<=n;i++)
            if(!used[i])
            {
                getsub(i,0,i);
                val[i]=cal(i);
                v2.push_back(val[i]);
            }
        sort(v2.begin(),v2.end());
        for(int i=1;i<=n;i++)
            if(sub[i].size())
            {
                ull t1=0,t2=0,t3=0;
                for(int j=0,flag=0;j<v2.size();j++)
                {
                    if(v2[j]==val[i]&&!flag)
                        {flag=1;continue;}
                    t1=t1*seed1+v2[j];
                }
                for(int j=0,t;j<sub[i].size();j++)
                {
                    vis[t=sub[i][j]]=1;
                    v3.clear();v3.push_back(node(1,0,0));
                    for(int k=0;k<v[t].size();k++)
                        v3.push_back(node(cal(v[t][k]),v[t][k],0));
                    sort(v3.begin(),v3.end());
                    for(int k=1;k<v3.size();k++)
                    {   
                        t2=0;
                        for(int w=0;w<v3.size();w++)
                            if(w!=k)t2=t2*seed1+v3[w].v;
                        if(id==0)t3=v3[k].v*seed2*seed2+t2*seed2+t1;
                        else t3=v3[k].v*seed2*seed2+t1*seed2+t2;
                        a[id].push_back(node(t3,v3[k].x,t));
                    }
                    t2=0;
                    for(int w=0;w<v3.size();w++)
                        t2=t2*seed1+v3[w].v;
                    if(id==0)t3=t2*seed2+t1;
                    else t3=t1*seed2+t2;
                    a[id].push_back(node(t3,0,t));
                    vis[t]=0;
                }
            }
        sort(a[id].begin(),a[id].end());
    }
    void print()
    {
        puts("YES");
        for(int i=1;i<=n;i++)
            for(int j=0,t;j<v[i].size();j++)
                if((t=v[i][j])<i)
                    printf("%d %d\n",i,t);
    }
}tr[2],ans;
int relabel(int x,int y,int type)
{
    int ret=pos[type][x]=++label;
    for(int i=0,t;i<tr[type].v[x].size();i++)
        if((t=tr[type].v[x][i])!=y)
        {
            int t1=relabel(t,x,type);
            ans.v[ret].push_back(t1);
            ans.v[t1].push_back(ret);
        }
    return ret;
}
void solve()
{
    scanf("%d%*d",&n);
    for(int i=0;i<=1;i++)
        tr[i].solve(i);

    for(int i=0,now=0;i<a[0].size();i++)
    {
        while(now<a[1].size()&&a[1][now].v<a[0][i].v)now++;
        if(now<a[1].size()&&a[1][now].v==a[0][i].v)
        {
            relabel(a[0][i].y,0,0);
            int t=label;
            relabel(a[1][now].y,a[1][now].x,1);
            ans.vis[pos[0][a[0][i].y]]=1;
            int t1=pos[1][a[1][now].y];
            ull v1=tr[1].cal(a[1][now].y);
            for(int j=1;j<=t;j++)
            {
                ans.v[j].push_back(t1);
                ans.v[t1].push_back(j);
                if(ans.cal(t1)==v1)
                {
                    ans.print();
                    return;
                }
                ans.v[j].pop_back();
                ans.v[t1].pop_back();
            }
        }
    }
    puts("NO");
}
int main()
{
    //freopen("tt.in","r",stdin);
    scanf("%d",&T);
    while(T--)
    {
        label=0;
        memset(pos,0,sizeof(pos));
        a[0].clear();a[1].clear();
        ans.init();
        solve();
    }
    return 0;
}