[ 斯特林数 ] Codeforces961G Partitions

显然最终答案中 $w_i$ 的系数是相等的。我们尝试计算 $w_i$ 的系数。
因为 $i$ 和 $j$ 在同一集合中会使 $w_i，w_j$ 的系数加 $1$ ，所以 $w_i$ 的系数可以表示成

$$\sum_{j=1}^n g_{i,j}$$
其中 $g_{i,j}$ 表示 $i$ 与 $j$ 在同一集合内的概率。
利用第二类斯特林数， $g_{i,i}=S(n,k),g_{i,j,i\neq j}=S(n-1,k)$ 。
求出 $S(n,k)$ 和 $S(n-1,k)$ 就好了。

#include<bits/stdc++.h>
using namespace std;
const int N=200010;
const int P=1000000007;
int k,n,m,x;
int s1[N],s2[N];
int Ans;
int Pow(int x,int y) {
    int Ans=1;
    for(;y;y>>=1,x=1ll*x*x%P)
        if(y&1) Ans=1ll*Ans*x%P;
    return Ans;
}
void Add(int& x,int y) {
    x=(x+y)%P;
}
void init() {
    s1[0]=s1[1]=s2[0]=s2[1]=1;
    for(int i=2;i<=k;i++) s2[i]=1ll*s2[P%i]*(P-P/i)%P;
    for(int i=2;i<=k;i++) s1[i]=1ll*s1[i-1]*i%P,s2[i]=1ll*s2[i-1]*s2[i]%P;
}
int C(int n,int m) {
    return 1ll*s1[n]*s2[m]%P*s2[n-m]%P;
}
int Calc() {
    int s=0,t=1;
    for(int i=0;i<k;i++,t*=-1) Add(s,1ll*t*C(k,i)*Pow(k-i,n)%P);
    s=1ll*s*s2[k]%P;
    int S=0;t=1;
    for(int i=0;i<k;i++,t*=-1) Add(S,1ll*t*C(k,i)*Pow(k-i,n-1)%P);
    S=1ll*S*s2[k]%P;
    Add(s,1ll*(n-1)*S%P);
    return (s+P)%P;
}
int main() {
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) scanf("%d",&x),Add(Ans,x);
    init();
    cout<<1ll*Ans*Calc()%P<<endl;
    return 0;
}