Codeforces 1097G

根本想不到

CF1097G

---

题意

给出一棵树，定义f(S)为用最少的边连通点集$ S$的边数

求$ \sum\limits f(S)^k$

$ n \leq 10^5 k \leq 200$

---

题解

假设$ k=1$有一个清真的树形$ DP$

在点集的$ LCA$处统计答案即可

对于$ k>1$根据二项式定理

可以$ O(nk^2)$完成转移

但这是过不去的

考虑

$$ x^k=\sum_{i=0}^k \binom{x}{i}S(k,i)i!$$

其中$ S(i,j)$表示第二类斯特林数

拆开组合数得

$$ x^k=\sum_{i=0}^k \frac{x!}{(x-i)!}S(k,i)$$

因此我们只要维护所有的下降幂就可以还原出$ x^k$

诶等等...这复杂度还是$ nk^2$的啊...

冷静分析一下，假设当前选取的边集大小不超过$ k$那下降幂为$ 0$

因此我们只需要枚举到$ min(当前非0下降幂的长度,k)$即可

根据树上背包的复杂度分析，其实是$ O(nk)$的

---

代码

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#define p 1000000007
#define rt register int
#define ll long long
using namespace std;
inline ll read(){
    ll x=;char zf=;char ch=getchar();
    while(ch!='-'&&!isdigit(ch))ch=getchar();
    if(ch=='-')zf=-,ch=getchar();
    while(isdigit(ch))x=x*+ch-'',ch=getchar();return x*zf;
}
void write(ll y){if(y<)putchar('-'),y=-y;if(y>)write(y/);putchar(y%+);}
void writeln(const ll y){write(y);putchar('\n');}
int k,m,n,x,y,z,cnt;
vector<int>e[];
int S[][],f[][],mi[],sz[],ans[];
void dfs(int x,int pre){
    f[x][]=sz[x]=;
    for(auto v:e[x])if(v!=pre){
        dfs(v,x);
        for(rt i=min(sz[v]-,k-);i>=;i--){
            int val=f[v][i];
            if(!i)val=1ll*val*(-mi[sz[v]])%p;
            (ans[i+]+=1ll*val*(-mi[n-sz[v]])%p)%=p;
            (f[v][i+]+=val)%=p;
        }
        for(rt i=min(sz[x]-,k);i>=;i--)
        for(rt j=;j<=sz[v]&&i+j<=k;j++){
            const int val=1ll*f[x][i]*f[v][j]%p;
            if(i)(ans[i+j]+=val)%=p;
            (f[x][i+j]+=val)%=p;
        }
        sz[x]+=sz[v];
    }
}
#define inv2 500000004
int main(){
    n=read(),k=read();
    S[][]=;
    for(rt i=;i<=k;i++)
    for(rt j=;j<=i;j++)S[i][j]=(S[i-][j-]+1ll*S[i-][j]*j%p)%p;
    for(rt i=;i<n;i++){
        x=read();y=read();
        e[x].push_back(y);
        e[y].push_back(x);
    }
    mi[]=;
    for(rt i=;i<=n;i++)mi[i]=1ll*mi[i-]*inv2%p;
    dfs(,);
    int ret=,jc=;
    for(rt i=;i<=k;i++)(ret+=1ll*S[k][i]*jc%p*ans[i]%p)%=p,jc=1ll*jc*(i+)%p;
    for(rt i=;i<=n;i++)ret=2ll*ret%p;
    cout<<(ret+p)%p;
    return ;
}