bzoj 3162: 独钓寒江雪 树哈希+树形dp

题意

给出一棵无标号无根树，问本质不同的最大独立集数量。答案模1e9+7。
n<=500000

分析

对于一般的情况，我们可以先找出树的重心作为根，然后进行树形dp。这样做有什么好处呢？通过根的性质不难发现以一个点为根时它父亲所在的子树肯定不会跟它的其他子树同构。那么就可以愉快的进行树形dp了。
十分显然的最大独立集求法：f[i,0/1]表示是否选i的最大独立集。
对于它的所有子树，把所有同构的分别放在一起算，方案的话就是一个组合数就没了。

十分愉快地码完后，我发现过不了样例三，因为这棵树有两个根。
于是对于两个根的情况，我们只要新加一个点连接两个根然后再dp即可。

又十分愉快地码完后，我发现还是过不了，因为被卡树哈希了，于是就去get了一波大爷的树哈希姿势。
大概是这样子的：

for (int i=1;i<=tot;i++) (((hash[x]*=BASE)+=tmp[i].first)^=tmp[i].first)+=tmp[i].first;

用大爷的话来说，这个哈希是越乱越好。
然后就可以过了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=500005;
const int MOD=1000000007;
const int BASE=233333333;

int n,last[N],cnt,val[N],size[N],f[N],g[N],root,pow[N],ny[N],jc[N],r1,r2;
unsigned long long hash[N];
struct edge{int to,next;}e[N*2];
pair<unsigned long long,int> tmp[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int ksm(int x,int y)
{
    int ans=1;
    while (y)
    {
        if (y&1) ans=(LL)ans*x%MOD;
        x=(LL)x*x%MOD;y>>=1;
    }
    return ans;
}

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void get_root(int x,int fa)
{
    size[x]=1;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa) continue;
        get_root(e[i].to,x);
        size[x]+=size[e[i].to];
        val[x]=max(val[x],size[e[i].to]);
    }
    val[x]=max(val[x],n-size[x]);
    if (!root||val[x]<val[root]) root=x;
}

void dp(int x,int fa)
{
    hash[x]=1;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa||x==r1&&e[i].to==r2||x==r2&&e[i].to==r1) continue;
        dp(e[i].to,x);
    }
    int tot=0;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa||x==r1&&e[i].to==r2||x==r2&&e[i].to==r1) continue;
        tmp[++tot]=make_pair(hash[e[i].to],e[i].to);
    }
    sort(tmp+1,tmp+tot+1);
    for (int i=1;i<=tot;i++) (((hash[x]*=BASE)+=tmp[i].first)^=tmp[i].first)+=tmp[i].first;
    f[x]=g[x]=1;
    int l=1;
    while (l<=tot)
    {
        int r=l;
        while (r<tot&&tmp[r+1].first==tmp[r].first) r++;
        int s=r-l+1,c=ny[s],w=g[tmp[l].second]+s-1;
        for (int i=0;i<s;i++) c=(LL)c*(w-i)%MOD;
        f[x]=(LL)f[x]*c%MOD;
        c=ny[s];w=(g[tmp[l].second]+f[tmp[l].second]+s-1)%MOD;
        for (int i=0;i<s;i++) c=(LL)c*(w-i)%MOD;
        g[x]=(LL)g[x]*c%MOD;
        l=r+1;
    }
}

int main()
{
    n=read();
    pow[0]=jc[0]=ny[0]=1;
    for (int i=1;i<=n+1;i++) pow[i]=(LL)pow[i-1]*n%MOD,jc[i]=(LL)jc[i-1]*i%MOD;
    ny[n+1]=ksm(jc[n+1],MOD-2);
    for (int i=n;i>=1;i--) ny[i]=(LL)ny[i+1]*(i+1)%MOD;
    for (int i=1;i<n;i++)
    {
        int x=read(),y=read();
        addedge(x,y);
    }
    get_root(1,0);
    for (int i=1;i<=n;i++)
        if (val[i]==val[root]) r2=r1,r1=i;
    if (r2) n++,addedge(n,r1),addedge(n,r2),root=n;
    dp(root,0);
    if (!r2) printf("%d",((f[root]+g[root])%MOD+MOD)%MOD);
    else if (hash[r1]!=hash[r2])
    {
        int ans=g[root];
        (ans-=(LL)f[r1]*f[r2]%MOD)%=MOD;
        printf("%d",(ans+MOD)%MOD);
    }
    else
    {
        int ans=g[root];
        int s=2,c=ny[s],w=f[r2]+s-1;
        for (int i=0;i<s;i++) c=(LL)c*(w-i)%MOD;
        (ans-=c)%=MOD;
        printf("%d",(ans+MOD)%MOD);
    }
    return 0;
}