[atARC087F]Squirrel Migration

对这棵树重心情况分类讨论：

1.若这棵树存在两个重心，分别记作$x$和$y$，如果将$(x,y)$断开，两棵子树大小都相同（都为$\frac{n}{2}$），此时$p_{i}$与$i$必然不同属于一个连通块中，证明如下：

考虑若$p_{i}$与$i$在一个连通块中，则必然有$p_{j}$和$j$也在同一个连通块中且与$i$不同，将其交换一定更优

将距离拆为两颗子树内部+$(x,y)$，即有$mx=2\sum dep_{i}(以(x,y)为根)+n$，方案数为$(\frac{n}{2}!)^{2}$

2.若这棵树仅有1个重心，类似于Distance Matching，若以重心为根，$mx=2\sum dep_{i}$

问题相当于要求任意$i$和$p_{i}$不在重心的同一个儿子中，考虑容斥，令集合$S$表示$i$和$p_{i}$在重心的同一个儿子中的$i$，$f_{S}$表示对应方案数

对$S$和$S$以外的的点分别计算（再相乘）：

1.对$S$以外，考虑$i\in S$，其实可以将$p_{i}$理解为$i$，换言之将$p_{j}=i$的位置改为$p_{j}=p_{i}$即可，因此剩下的点任意排列，方案数为$(n-|S|)!$

2.对$S$以内，即从子树中选$i$个，即为$\prod_{son}a_{son}!c(sz_{son},a_{son})$（$son$表示重心的儿子，$a_{son}$表示son子树内所选的节点个数）

由于$|S|=\sum_{son}a_{son}$，因此答案仅与$a_{i}$有关，对应答案为$(n-\sum_{son}a_{son})!\prod_{son}a_{son}!c(sz_{son},a_{son})^{2}$（对应的$|S|$有$\prod_{son}c(sz_{son},a_{son})$种）

令$f_{i}$表示当$\sum_{son}a_{son}=i$时$\prod_{son}c(sz_{son},a_{son})^{2}$的和，dp转移即可，时间复杂度为$o(n^{2})$（枚举子树和$a_{i}$的总量为$o(n)$），还可以用分治fft优化到$o(n\log^{2}n)$

#include<bits/stdc++.h>
using namespace std;
#define N 5005
#define mod 1000000007
struct ji{
    int nex,to;
}edge[N<<1];
vector<int>v;
int E,n,m,x,y,ans,head[N],sz[N],fac[N],inv[N],f[N][N];
int sqr(int n){
    return 1LL*n*n%mod;
}
int c(int n,int m){
    return 1LL*fac[n]*inv[m]%mod*inv[n-m]%mod;
}
void add(int x,int y){
    edge[E].nex=head[x];
    edge[E].to=y;
    head[x]=E++;
}
void dfs(int k,int fa){
    int mx=0;
    sz[k]=1;
    for(int i=head[k];i!=-1;i=edge[i].nex)
        if (edge[i].to!=fa){
            dfs(edge[i].to,k);
            mx=max(mx,sz[edge[i].to]);
            sz[k]+=sz[edge[i].to];
        }
    if (max(mx,n-sz[k])<=n/2){
        if (!x)x=k;
        else y=k;
    }
}
int main(){
    fac[0]=inv[0]=inv[1]=1;
    for(int i=1;i<N-4;i++)fac[i]=1LL*fac[i-1]*i%mod;
    for(int i=2;i<N-4;i++)inv[i]=1LL*(mod-mod/i)*inv[mod%i]%mod;
    for(int i=1;i<N-4;i++)inv[i]=1LL*inv[i]*inv[i-1]%mod;
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    for(int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    x=y=0;
    dfs(1,0);
    if (y){
        printf("%d",sqr(fac[n/2]));
        return 0;
    }
    dfs(x,0);
    v.push_back(0);
    for(int i=head[x];i!=-1;i=edge[i].nex)v.push_back(sz[edge[i].to]);
    m=v.size()-1;
    f[0][0]=1;
    int s=0;
    for(int i=1;i<=m;i++){
        for(int j=0;j<=s;j++)
            for(int k=0;k<=v[i];k++)
                f[i][j+k]=(f[i][j+k]+1LL*fac[k]*sqr(c(v[i],k))%mod*f[i-1][j])%mod;
        s+=v[i];
    }
    assert(s==n-1);
    for(int i=0;i<n;i++){
        int s=1LL*f[m][i]*fac[n-i]%mod;
        if (i&1)s=mod-s;
        ans=(ans+s)%mod;
    }
    printf("%d",ans);
}