[组合斯特林数] Codeforces 932E. Team Work

最新推荐文章于 2021-11-12 14:07:12 发布

LowestJN

最新推荐文章于 2021-11-12 14:07:12 发布

阅读量542

点赞数

分类专栏：斯特林数

本文链接：https://blog.csdn.net/coldef/article/details/79330291

版权

斯特林数专栏收录该内容

2 篇文章 0 订阅

订阅专栏

a n s = \sum i = 1 n (n i) i k

$ans=\sum_{i=1}^n{n\choose i}i^k$

用斯特林数展开 $i^k$

a n s = \sum i = 1 n (n i) \sum j = 1 k S (k, j) A (i, j)

$ans=\sum_{i=1}^n{n\choose i}\sum_{j=1}^kS(k,j)A(i,j)$

= \sum j = 1 k S (k, j) \sum i = 1 n (n i) A (i, j)

$=\sum_{j=1}^kS(k,j)\sum_{i=1}^n{n\choose i}A(i,j)$

考虑 $\sum_{i=1}^n{n\choose i}A(i,j)$ 的组合意义

$n$ 个物品里选 $j$ 个的排列，剩下的 $n-j$ 可选可不选

那么 $\sum_{i=1}^n{n\choose i}A(i,j)=A(n,j)\times 2^{n-j}$

所以答案就是 $\sum_{j=1}^kS(k,j)\times A(n,j)\times 2^{n-j}$

$O(k^2)$

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

const int N=5010,P=1e9+7;

int S[N][N];

inline int Pow(int x,int y){
  int ret=1;
  for(;y;y>>=1,x=1LL*x*x%P) if(y&1) ret=1LL*ret*x%P;
  return ret;
}

inline int A(int n,int j){
  int ret=1;
  for(int i=0;i<j;i++)
    ret=1LL*ret*(n-i)%P;
  return ret;
}

int main(){
  S[0][0]=1;
  for(int i=1;i<=5000;i++){
    for(int j=1;j<=i;j++)
      S[i][j]=(S[i-1][j-1]+1LL*j*S[i-1][j])%P;
  }
  int ans=0;
  int n,k; scanf("%d%d",&n,&k);
  for(int j=1;j<=k && j<=n;j++)
    ans=(ans+1LL*S[k][j]*A(n,j)%P*Pow(2,n-j))%P;
  printf("%d\n",ans);
  return 0;
}