【CF1097G】Vladislav and a Great Legend

最新推荐文章于 2021-05-31 21:44:49 发布

HOWARLI

最新推荐文章于 2021-05-31 21:44:49 发布

阅读量554

点赞数

分类专栏： # 斯特林数文章标签：斯特林数

本文链接：https://blog.csdn.net/HOWARLI/article/details/86645331

版权

斯特林数专栏收录该内容

2 篇文章 0 订阅

订阅专栏

Description

个给出一棵树，定义 $f (S)$ 表示对于点集 $S$ ，需要把点集中的所有点连接起来至少需要的边数，
现在对于所有S求 $f(S)^k$ 之和，也就是求 $\sum f(S)^k$

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

Solution

~~我又双叒叕来学斯特林数了~~

看到次方自然的用斯特林数拆一下：
$f(S)^k=\sum_{i=0}^kS(k,i)i!C_{f(S)}^i$
所以：
$Ans=\sum_{i=0}^kS(k,i)i!\sum C_{f(S)}^i$
直接维护后面那个东西即可，

设 $f_{x,i}$ 表示点X的子树中，选了i条边的选点方案数，这里我们强制要求x到父亲的边必须存在（即x上面一定选了点）

初始转态： $f_{x,0}=2$ ，表示点x选不选（边数均为0），
转移直接把两个卷积起来即可，最后再加上x到父亲的边，要注意当x子树中没有点时要减掉，因为边数还是0，

关于答案的计算，我们每次统计子树中经过点x的答案，
要注意一下：如果是跨子树的情况，那么点x可以选可以不选，但如果只有一棵子树，那么点x必选，

复杂度： $O(nk^2)$
~~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;
}