BZOJ 3162：独钓寒江雪

BZOJ 3162：独钓寒江雪

题目传送门
PS：唉，十年没有写博客了。。。。

【问题描述】

　　题目过长，没有描述。。
　　大意就是，求一个诡异的雪花有多少种不同的01染色方法（1不能和1直接相连，旋转之后相同的算同一种）。

【解题思路】

　　假设，即使旋转相同也算做不同。那么，这就是一个很easy的树形DP。
　　f[x][1]=f[x][1]*f[son][0];
　　f[x][0]=f[x][0]*(f[son][0]+f[son][1]);
　　但是，如何知道树的形态是否一致呢？
　　可以给树编码。HASH，假如hash值一致，就认为是一样的。
　　HASH当然是越诡异越好。保证不同的树hash值不一致，相同的树hash值一致。
　　但，树的中心怎么找？
　　也只有找出中心，才好判断树是否有对称部分（若对称则会算重）。
　　以树的中心为第一层。才能保证对称的树在同一个father下。
　　找出树的直径，直径的中点就是中心。若中点在边上，则新建一个点作为中心。

　　若是相同的话，就等于有x种小球，n个箱子。每个箱子放一种小球，箱子是一样的，求有多少种放法？
　　就是
　　1<=a1<=x
　　a1<=a2<=x
　　a2<=a3<=x
　　……
　　an-1<=an<=x
　　a序列的方案数。
　　隔板法：C(n,n+x-1)
　　预处理一发阶乘，阶乘的逆元。（我并没有这么干……）

　　找一发重心AC了。（找不出证明，估计是水过的……重心比中心好写~\（^。^）/~ ）

　　更具体的请参照：vfleaking的博客
【代码】

#include<bits/stdc++.h>

#define imax(a,b) ((a>b)?(a):(b))
#define imin(a,b) ((a<b)?(a):(b))

using namespace std;

typedef long long ll;

const int N=500010;
const int inf=1e9+50;
const int mods=1e9+7;

int n,root,r1,r2,sum;
int siz[N],mp;
int to[N<<1],ne[N<<1],h[N<<1],tt;
int now[N],top;
unsigned long long H[N];
ll f[N][2],inv[N];

void read(int &x)
{
    x=0; char ch=getchar(); int f=1;
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-1;
    for(; isdigit(ch);ch=getchar()) x=(x<<3)+(x<<1)+ch-'0';
    x*=f;
}

void addedge(int a,int b) { to[++tt]=b; ne[tt]=h[a]; h[a]=tt; }

bool cmp(int A,int B) { return (H[A]>H[B]); }

void getroot(int x,int fa)
{
    int dd=0; siz[x]=1;
    for(int p=h[x];p;p=ne[p])
    {
        int v=to[p];
        if(v==fa) continue;
        getroot(v,x);
        siz[x]+=siz[v];
        dd=imax(dd,siz[v]);
    }
    dd=imax(dd,sum-siz[x]);
    if(dd<mp) root=x,mp=dd,r1=x,r2=0; else
    if(dd==mp) r2=x;
}

ll col(int n1,int n2)
{
    int n=n1+n2-1;
    ll tmp=1ll; n%=mods;
    for(int i=1;i<=n1;i++)
    tmp=(tmp*(n-i+1))%mods*inv[i]%mods;
    return tmp;
}

void dp(int x,int fa)
{
    f[x][0]=f[x][1]=1;
    for(int p=h[x];p;p=ne[p])
    if(to[p]!=fa) dp(to[p],x);
    top=0;
    for(int p=h[x];p;p=ne[p])
    if(to[p]!=fa) now[++top]=to[p];
    sort(now+1,now+1+top,cmp);
    int r;
    for(int i=1;i<=top;i=r)
    {
        for(r=i+1;r<=top && H[now[i]]==H[now[r]];r++);
        f[x][0]=(f[x][0]*col(r-i,f[now[i]][0]+f[now[i]][1]))%mods;
        f[x][1]=(f[x][1]*col(r-i,f[now[i]][0]))%mods;
    }
    H[x]=19;
    for(int i=1;i<=top;i++)
    H[x]=H[x]*9875321+H[now[i]]*17;
}

int main()
{
    read(n); tt=1; inv[1]=1ll;
    for(int i=2;i<=n;i++) inv[i]=inv[mods%i]*(mods-mods/i)%mods;
    for(int i=1;i<n;i++) 
    {
        int a,b; read(a); read(b);
        addedge(a,b); addedge(b,a);
    }
    sum=n; root=0; mp=inf;
    getroot(1,0);
    if(r2)
    {
        for(int p=h[r1];p;p=ne[p])
        if(to[p]==r2)
        {
            to[p^1]=to[p]=n+1;
            root=n+1;
        }
        addedge(root,r1); addedge(root,r2);
    }
    dp(root,0);
    if(!r2) sum=(f[root][0]+f[root][1])%mods; else
    {
        if(H[r1]==H[r2])
        sum=(f[r1][0]*f[r2][1]%mods+col(2,f[r1][0]))%mods; else
        sum=(f[r1][0]*f[r2][1]%mods+f[r1][1]*f[r2][0]%mods+f[r1][0]*f[r2][0]%mods)%mods;
    }
    printf("%lld\n",sum);
    return 0;
}