luogu P4827 [国家集训队] Crash 的文明世界

背景：

自己的模拟赛自己的防$\text{AK}$题防住了自己。
$\text{NTT}$不熟啊。

题目传送门：

https://www.luogu.org/problem/P4827

题意：

一棵树，对于每一个点$x$求：${\sum }_{i=1}^n\text{dis}(x,i{)}^k$。

思路$1$：

显然可以用点分治，时间复杂度：$\Theta (n\log n\log k)$。
这差不多是模板题了。
懒人模式，代码就不打了。

思路$2$：

看到这种$k$次方的式子，容易想到伯努利数或者第二类斯特林数。
显然后者应用范围更广。
假设$x$为当前的根。
$$\sum _{i=1}^n\text{dis}(x,i{)}^k$$

因为$\text{dis}(x,i{)}^k={\sum }_{j=0}^k{S}_{k,j}{C}_{\text{dis}(x,i)}^{j}j!$，所以有：

$$=\sum _{i=1}^n\sum _{j=0}^k{S}_{k,j}{C}_{\text{dis}(x,i)}^{j}j!$$

$$=\sum _{j=0}^k{S}_{k,j}j!\sum _{i=1}^n{C}_{\text{dis}(x,i)}^j$$

$$=\sum _{j=0}^k{S}_{k,j}j!\sum _{i=1}^n{C}_{\text{dis}(x,i)-1}^j+C_{\text{dis}(x,i)-1}^{j-1}$$

考虑用$\text{dp}$来处理$C_{\text{dis}(x,i)}^j$，设$f_{x,j}=C_{\text{dis}(x,i)}^j$，因为$i$和$i$的儿子的距离差为$1$，因此有转移方程：
$$f_{x,j}=\sum _{y\in i.son}{f}_{y,j}+f_{y,j-1}$$

每一次转移时考虑换根即可。
$x$（父亲）转移到$y$（儿子），将$x$在$y$中的贡献减去，再新加入$x$对$y$的新贡献即可。
时间复杂度：$\Theta (nk)$。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define mod 10007
using namespace std;
	int n,k,len,ans;
	int last[50010],now[50010],fac[50010],s[160][160],f[50010][160];
	struct node{int x,y,next;} a[100010];
void ins(int x,int y)
{
	a[++len]=(node){x,y,last[x]}; last[x]=len;
}
void dfs1(int x,int fa)
{
	f[x][0]=1;
	for(int i=last[x];i;i=a[i].next)
	{
		int y=a[i].y;
		if(y==fa) continue;
		dfs1(y,x);
		f[x][0]=(f[x][0]+f[y][0])%mod;
		for(int j=1;j<=k;j++)
			f[x][j]=(f[x][j]+f[y][j]+f[y][j-1])%mod;
	}
}
void dfs2(int x,int fa)
{
	for(int i=last[x];i;i=a[i].next)
	{
		int y=a[i].y;
		if(y==fa) continue;
		now[0]=(f[x][0]-f[y][0]+mod)%mod;
		for(int j=1;j<=k;j++)
			now[j]=(f[x][j]-f[y][j]-f[y][j-1]+mod*2)%mod;
		f[y][0]=(f[y][0]+now[0])%mod;
		for(int j=1;j<=k;j++)
			f[y][j]=(f[y][j]+now[j-1]+now[j])%mod;
		dfs2(y,x);
	}
}
void init()
{
	fac[0]=1;
	s[0][0]=1;
	for(int i=1;i<=k;i++)
	{
		fac[i]=fac[i-1]*i%mod;
		for(int j=1;j<=i;j++)
			s[i][j]=(s[i-1][j-1]+j*s[i-1][j]%mod)%mod;
	}
}
int main()
{
	int x,y;
	scanf("%d %d",&n,&k);
	init();
	for(int i=1;i<n;i++)
	{
		scanf("%d %d",&x,&y);
		ins(x,y),ins(y,x);
	}
	dfs1(1,0);
	dfs2(1,0);
	for(int i=1;i<=n;i++)
	{
		int tot=0;
		for(int j=1;j<=k;j++)
			tot=(tot+s[k][j]*fac[j]%mod*f[i][j]%mod)%mod;
		printf("%d\n",tot);
	}
}