codeforces 932E Team Work 题解

题目传送门

题目大意： 给出 $n,k$，求 ${\sum }_{i=1}^n{C}_n^i\times i^k$。

题解

看到 $i^k$ 就想到用第二类斯特林数来展开了。

考虑这个东西的意义：将 $k$ 个不同的小球放到 $i$ 个不同的盒子中的方案数。那么可以枚举一个 $j$，表示有 $j$ 个盒子不是空的（剩下的 $i-j$ 个是空的），那么方案数为 $C_i^j\times S(k,j)\times j!$，$C_i^j$ 表示从 $i$ 个盒子中选出 $j$ 个不为空的盒子，然后将 $k$ 个不同的小球放进去的方案数就是第二类斯特林数 $S(k,j)$，由于盒子是不一样的，所以还要乘上 $j!$。

然后大力推一下柿子：
$$\begin{array}{rl}& \sum _{i=1}^n{C}_n^i\times i^k\\ & =\sum _{i=1}^n{C}_n^i\sum _{j=1}^n{C}_i^j\times S(k,j)\times j!\\ & =\sum _{j=1}^{n}S(k,j)\times j!\sum _{i=1}^n{C}_n^i{C}_i^j\\ & =\sum _{j=1}^{n}S(k,j)\times j!\sum _{i=1}^n{C}_n^j{C}_{n-j}^{i-j}\\ & =\sum _{j=1}^{n}S(k,j)\times j!\times C_n^j\sum _{i=j}^n{C}_{n-j}^{i-j}\\ & =\sum _{j=1}^{n}S(k,j)\times j!\times \frac{n!}{j!(n-j)!}\times 2^{n-j}\\ & =\sum _{j=1}^{n}S(k,j)\times \frac{n!}{(n-j)!}\times 2^{n-j}\end{array}$$

此时其实已经可以求解了，虽然枚举的上限是 $n$，但是可以发现，当 $j$ 大于 $k$ 时 $S(k,j)=0$，所以不需要枚举到 $j>k$ 的部分，那么先 $k^2$ 预处理 $S(k,j)$，就可以求出答案了。

代码如下：

#include <cstdio>
#define maxn 5010
#define mod 1000000007

int n,k,S[maxn][maxn],ans=0;
int ksm(int x,int y)
{
	int re=1;
	while(y)
	{
		if(y&1)re=1ll*re*x%mod;
		x=1ll*x*x%mod;y>>=1;
	}
	return re;
}
int min(int &x,int &y){return x<y?x:y;}

int main()
{
	scanf("%d %d",&n,&k);
	S[0][0]=1;
	for(int i=1;i<=k;i++)for(int j=1;j<=i;j++)
	S[i][j]=(1ll*S[i-1][j]*j%mod+S[i-1][j-1])%mod;
	for(int i=1,prod=n;i<=min(n,k);prod=1ll*prod*(n-i)%mod,i++)
	ans=(ans+1ll*S[k][i]*prod%mod*ksm(2,n-i)%mod)%mod;
	printf("%d",ans);
}