正題

題目鏈接:https://www.luogu.com.cn/problem/CF961G

---

題目大意

$n$個物品分成$k$個組，每個物品權值為$w_i$。一個子集$S$的權值為$|S|{\sum }_{x\in S}{w}_x$。

求所有劃分方法的權值和。

---

解題思路

考慮對于每個數$w_i$的貢獻，可以看為同一集合內每一個數都對其進行過貢獻。$w_i$對自身的貢獻為$\left\{\begin{array}{c}n\\ k\end{array}\right\}$（也就是每一種劃分方案都是）。對于$j=i$的貢獻，我們將$w_i$和$w_j$綁定在一個組中，方案數就是$\left\{\begin{array}{c}n?1\\ k\end{array}\right\}$。也就是對于每一個會被計算$\left\{\begin{array}{c}n\\ k\end{array}\right\}+(n?1)\left\{\begin{array}{c}n?1\\ k\end{array}\right\}$

考慮用斯特林數的通項計算（先是通項的推導）。
定義$f(x)=x^n,g(x)=\left\{\begin{array}{c}n\\ x\end{array}\right\}x!$
那么有$$f(x)=x^n=\sum _{k=0}^x\left(\genfrac{}{}{0ex}{}{x}{k}\right)\left\{\begin{array}{c}n\\ k\end{array}\right\}k!=\sum _{k=0}^x\left(\genfrac{}{}{0ex}{}{x}{k}\right)g(k)$$
根據二項式反演有$$g(k)=\sum _{i=1}^k(?1{)}^{k?i}(\genfrac{}{}{0ex}{}{n}{i})f(i)$$
也就是$$\left\{\begin{array}{c}n\\ k\end{array}\right\}k!=\sum _{i=1}^k(?1{)}^{k?i}(\genfrac{}{}{0ex}{}{n}{i})i^k$$
化簡得$$\left\{\begin{array}{c}n\\ m\end{array}\right\}=\frac{1}{m!}\sum _{k=0}^m(?1{)}^k(\genfrac{}{}{0ex}{}{m}{k})(m?k{)}^n$$

然后計算即可。

---

$code$

#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll XJQ=1e9+7,N=2e5+10; ll n,m,ans,inv[N],fac[N]; ll power(ll x,ll b){ll ans=1;while(b){if(b&1)ans=ans*x%XJQ;x=x*x%XJQ;b>>=1;}return ans; } void init(){fac[0]=inv[0]=1;for(ll i=1;i<=m;i++){inv[i]=inv[i-1]*power(i,XJQ-2)%XJQ;fac[i]=fac[i-1]*i%XJQ;} } ll C(ll n,ll m) {return fac[n]*inv[n-m]%XJQ*inv[m]%XJQ;} ll S(ll n,ll m){ll ans=0,z=-1;for(ll i=0;i<=m;i++)z*=-1,(ans+=z*C(m,i)*power(m-i,n)%XJQ)%=XJQ;ans=ans*inv[m]%XJQ;return (ans+XJQ)%XJQ; } int main() {scanf("%lld%lld",&n,&m);init();ll k1=S(n,m),k2=S(n-1,m)*(n-1)%XJQ;for(ll i=1;i<=n;i++){ll x;scanf("%lld",&x);(ans+=x*(k1+k2)%XJQ)%=XJQ;}printf("%lld",ans); }

總結

以上是生活随笔為你收集整理的CF961G-Partitions【斯特林数】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。