Newcoder 141 B.Expected Number of Nodes（组合数学）

Description

给出一棵$n$个点的树，每次随机选$k$个不同的点，对于没选中的点，度数为$1$的点直接删除，度数为$2$的点则删除这个点以及这个点所有邻接点，持续上述过程直至所有点要么被选中要么度数超过$2$，问对于不同的$k$，剩余点数的期望值

Input

第一行一整数$n$表示点数，之后$n-1$行输入$n-1$条树边$(1\le n\le 5000)$

Output

输出$n$个整数，分别表示$k=1,...,n$时剩余点数的期望值，结果模$10^9+7$

Sample Input

4
1 2
2 3
3 4

Sample Output

1
2
3
4

Solution

剩余点数期望值即为每点不被删除的概率之和，根据点的度数不同分两种情况：

1.$du[u]\le 2$，则只要$u$点不被选则会被删掉，该点不被删除的概率为$\frac{C_{n-1}^{k-1}}{C_n^k}$

2.$du[u]>2$，假设以$u$为根，假设所选$k$个点在$u$的$x$个子树中，那么处理完$u$的所有子树后$u$的度为$x$，故只要$x\le 2$则$u$会被删掉，假设$u$有$m$棵子树，节点数分别为$s_1,...,s_m$，那么$u$不被删除的概率为

$1-\frac{1}{C_n^k}(\sum\limits_{i<j}(C_{s_i+s_j}^k-C_{s_i}^k-C_{s_j}^k)+\sum\limits_{i}C_{s_i}^k)=1-\frac{1}{C_n^k}(\sum\limits_{i<j}C_{s_i+s_j}^k+(2-m)\sum\limits_{i}C_{s_i}^k)$

第一种情况很好计算，统计度不超过$2$的点数直接计算即可，而对于第二种情况，由于要求出$k=1,...,n$的所有期望值，故直接计算时间复杂度$O(n^3)$，但是注意到我们可以对于所有度数超过$2$的点$u$把括号内$C_{i}^k$项前的系数$a_i$预处理出来，那么对于枚举的$k$，直接计算$\sum\limits_{i=1}^na_iC_i^k$即可，如此计算时间复杂度$O(n^2)$

Code

#include<cstdio>
#include<vector>
using namespace std;
typedef long long ll;
#define maxn 5005
#define mod 1000000007
int mul(int x,int y)
{
    ll z=1ll*x*y;
    return z-z/mod*mod;
}
int add(int x,int y)
{
    x+=y;
    if(x>=mod)x-=mod;
    return x;
}
int C[maxn][maxn];
int Pow(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1)ans=mul(ans,a);
        a=mul(a,a);
        b>>=1;
    }
    return ans;
}
void init(int n)
{
    C[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        C[i][0]=C[i][i]=1;
        for(int j=1;j<i;j++)C[i][j]=add(C[i-1][j-1],C[i-1][j]);
    }
}
int n,Size[maxn],a[maxn];
vector<int>e[maxn],s[maxn];
void dfs(int u,int fa)
{
    Size[u]=1;
    for(int i=0;i<e[u].size();i++)
    {
        int v=e[u][i];
        if(v==fa)continue;
        dfs(v,u);
        s[u].push_back(Size[v]);
        Size[u]+=Size[v];
    }   
}
int main()
{
    scanf("%d",&n);
    init(n);
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        e[u].push_back(v),e[v].push_back(u);
    }
    dfs(1,0);
    for(int i=2;i<=n;i++)s[i].push_back(n-Size[i]);
    int cnt=0;
    for(int i=1;i<=n;i++)
        if(s[i].size()>2)
        {
            cnt++;
            for(int j=0;j<s[i].size();j++)a[s[i][j]]=add(a[s[i][j]],mod+2-s[i].size());
            for(int j=0;j<s[i].size();j++)  
                for(int k=j+1;k<s[i].size();k++)
                    a[s[i][j]+s[i][k]]=add(a[s[i][j]+s[i][k]],1); 
        }
    for(int i=1;i<=n;i++)
    {
        int res=0;
        for(int j=i;j<=n;j++)res=add(res,mul(a[j],C[j][i]));
        res=add(cnt,mod-mul(res,Pow(C[n][i],mod-2)));
        res=add(res,mul(n-cnt,mul(C[n-1][i-1],Pow(C[n][i],mod-2))));
        printf("%d\n",res);
    }
    return 0;
}