Description
个给出一棵树,定义
f
(
S
)
f(S)
f(S)表示对于点集
S
S
S,需要把点集中的所有点连接起来至少需要的边数,
现在对于所有S求
f
(
S
)
k
f(S)^k
f(S)k之和,也就是求
∑
f
(
S
)
k
\sum f(S)^k
∑f(S)k
n ≤ 1 0 5 , k ≤ 200 n\leq 10^5,k\leq 200 n≤105,k≤200
Solution
我又双叒叕来学斯特林数了
看到次方自然的用斯特林数拆一下:
f
(
S
)
k
=
∑
i
=
0
k
S
(
k
,
i
)
i
!
C
f
(
S
)
i
f(S)^k=\sum_{i=0}^kS(k,i)i!C_{f(S)}^i
f(S)k=i=0∑kS(k,i)i!Cf(S)i
所以:
A
n
s
=
∑
i
=
0
k
S
(
k
,
i
)
i
!
∑
C
f
(
S
)
i
Ans=\sum_{i=0}^kS(k,i)i!\sum C_{f(S)}^i
Ans=i=0∑kS(k,i)i!∑Cf(S)i
直接维护后面那个东西即可,
设 f x , i f_{x,i} fx,i表示点X的子树中,选了i条边的选点方案数,这里我们强制要求x到父亲的边必须存在(即x上面一定选了点)
初始转态:
f
x
,
0
=
2
f_{x,0}=2
fx,0=2,表示点x选不选(边数均为0),
转移直接把两个卷积起来即可,最后再加上x到父亲的边,要注意当x子树中没有点时要减掉,因为边数还是0,
关于答案的计算,我们每次统计子树中经过点x的答案,
要注意一下:如果是跨子树的情况,那么点x可以选可以不选,但如果只有一棵子树,那么点x必选,
复杂度:
O
(
n
k
2
)
O(nk^2)
O(nk2)
CF跑得快2333
Code
#include <cstdio>
#include <algorithm>
#include <iostream>
#define fo(i,a,b) for(int i=a;i<=b;++i)
#define fod(i,a,b) for(int i=a;i>=b;--i)
#define efo(i,q) for(int i=A[q];i;i=B[i][0])
#define min(q,w) ((q)>(w)?(w):(q))
#define max(q,w) ((q)<(w)?(w):(q))
using namespace std;
typedef long long LL;
const int N=100500,M=202,mo=1e9+7;
int read(int &n)
{
char ch=' ';int q=0,w=1;
for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
if(ch=='-')w=-1,ch=getchar();
for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int m,n;
int B[2*N][2],A[N],B0;
int Si[N];
LL ans;
LL f[N][M],f1[N][M],Ans[M];
LL st1[M][M];
void link(int q,int w)
{
B[++B0][0]=A[q],A[q]=B0,B[B0][1]=w;
B[++B0][0]=A[w],A[w]=B0,B[B0][1]=q;
}
void dfs(int q,int fa)
{
f[q][0]=2;Si[q]=1;
efo(i,q)if(B[i][1]!=fa)
{
dfs(B[i][1],q);
for(int k=0;k<=m&&k<=Si[q];++k)
for(int j=0;j<=Si[B[i][1]]&&j+k<=m;++j)f[0][k+j]=(f[0][k+j]+f[q][k]*f[B[i][1]][j])%mo;
Si[q]+=Si[B[i][1]];
for(int j=0;j<=m&&j<=Si[q];++j)
{
f[q][j]=f[0][j],f[0][j]=0;
f1[q][j]=(f1[q][j]+f[B[i][1]][j])%mo;
}
}
fo(i,0,m)Ans[i]=(Ans[i]+f[q][i]-f1[q][i])%mo;
fod(i,m,1)f[q][i]=(f[q][i]+f[q][i-1])%mo;
--f[q][1];
}
int main()
{
freopen("!.in","r",stdin);
// freopen(".out","w",stdout);
int q,w;
read(n),read(m);
fo(i,2,n)read(q),read(w),link(q,w);
st1[0][0]=1;
fo(i,1,m)
{
fo(j,1,i)st1[i][j]=(st1[i-1][j-1]+st1[i-1][j]*j)%mo;
}
dfs(1,0);
ans=0;
LL t=1;
fo(i,0,m)
{
ans=(ans+st1[m][i]*t%mo*Ans[i])%mo;
t=t*(i+1LL)%mo;
}
printf("%lld\n",(ans+mo)%mo);
return 0;
}