Codeforces 961G Partitions 组合数学

题意

有n个元素，每个元素有一个权值$w_i$。定义全集的一个子集$S$的权值$w(S)=|S|\sum_{i\in S}w_i$。当把全集划分为若干个集合时，定义一种划分的权值为每个集合的权值和。现在要求所有把全集划分成$k$个非空集合方案的权值和。
$k\le n\le 2*10^5$

分析

考虑把每一个元素的贡献加起来。
不难发现每个元素的系数都是一样的，所以我们只要把系数求出来即可。
一开始的想法是枚举元素$i$所在的集合大小，设为$s$，那么系数就是

$$\sum_{s=1}^{n}s*C_{n-1}^{s-1}*S(n-s,k-1)$$
其中 $S(n,k)$表示第二类斯特林数。
现在的问题在于如何快速算后面的斯特林数，而当k固定时的斯特林数好像并没有什么快速算法，所以我们需要换一种做法。
注意到集合权值前面的系数是 $|S|$，这里可以看成集合中的每一个元素都会对 $w_i$做一次贡献，那么我们只要枚举每一个数对 $w_i$的贡献就好了。
首先 $i$对$w_i$系数的贡献显然是 $S(n,k)$，当 $j\not=i$时 $j$对$w_i$的贡献就是 $j$和$i$在同一个集合中的划分方案数，也就是 $S(n-1,k)$。
综上所述， $w_i$的系数实际上就是 $S(n,k)+(n-1)*S(-1,k)$。
那么只要用容斥的方法 $O(nlogn)$来算斯特林数即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=200005;
const int MOD=1000000007;

int n,k,jc[N],ny[N];

int ksm(int x,int y)
{
    int ans=1;
    while (y)
    {
        if (y&1) ans=(LL)ans*x%MOD;
        x=(LL)x*x%MOD;y>>=1;
    }
    return ans;
}

int S(int n,int m)
{
    int ans=0;
    for (int i=1;i<=m;i++)
        if ((m-i)&1) (ans+=MOD-(LL)ksm(i,n)*ny[i]%MOD*ny[m-i]%MOD)%=MOD;
        else (ans+=(LL)ksm(i,n)*ny[i]%MOD*ny[m-i]%MOD)%=MOD;
    return ans;
}

int main()
{
    scanf("%d%d",&n,&k);
    jc[0]=jc[1]=ny[0]=ny[1]=1;
    for (int i=2;i<=n;i++) jc[i]=(LL)jc[i-1]*i%MOD,ny[i]=(LL)(MOD-MOD/i)*ny[MOD%i]%MOD;
    for (int i=2;i<=n;i++) ny[i]=(LL)ny[i]*ny[i-1]%MOD;
    int c=(S(n,k)+(LL)(n-1)*S(n-1,k)%MOD)%MOD,ans=0;
    for (int i=1;i<=n;i++)
    {
        int x;scanf("%d",&x);
        (ans+=(LL)x*c%MOD)%=MOD;
    }
    printf("%d",ans);
    return 0;
}