传送门
题解:
首先我们可以直接知道答案为 ∑ s n , k ⋅ k m \sum{s_{n,k}\cdot k^m} ∑sn,k⋅km
用数学方法处理稍微麻烦了一点,考虑其组合意义。
设 x i x_i xi表示循环 i i i是否出现,则一个置换的权值为 ( ∑ x i ) m (\sum x_i)^m (∑xi)m
展开,考虑其中由某 k k k个项组成的项的系数,相当于是 m m m个物品划分到 k k k个带标号集合中的方案数,则系数为 S m , k k ! S_{m,k}k! Sm,kk!。而这里 x i x_i xi只可能为 0 / 1 0/1 0/1,则系数就是对答案的最后贡献。
考虑某 k k k个循环,如果它们同时出现,就会对答案有 S m , k k ! S_{m,k}k! Sm,kk!的贡献。
那么考虑这 k k k个以外剩下的循环,我们新加一个点 n + 1 n+1 n+1,剩下的循环按照字典顺序与 n + 1 n+1 n+1可以连成一个新的环,并且这样的方案是唯一的。所以选出前 k k k个环的方案数为 s n + 1 , k + 1 s_{n+1,k+1} sn+1,k+1。
则答案为 ∑ k s n + 1 , k + 1 S m , k k ! \sum_{k}s_{n+1,k+1}S_{m,k}k! ∑ksn+1,k+1Sm,kk!
代码:
#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const
cs int mod=1e9+7;
inline int add(int a,int b){a+=b-mod;return a+((a>>31)&mod);}
inline int dec(int a,int b){a-=b;return a+((a>>31)&mod);}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
return res;
}
inline void Inc(int &a,int b){a+=b-mod;a+=(a>>31)&mod;}
inline void Dec(int &a,int b){a-=b;a+=(a>>31)&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}
class CyclesNumber{
private:
static cs int N=1e5+7,M=3e2+7;
int s[N][M],S[M][M],fac[M];
inline void init_stirling(int n,int m){
s[0][0]=S[0][0]=fac[0]=fac[1]=1;
for(int re i=2;i<=m;++i)fac[i]=mul(fac[i-1],i);
for(int re i=1;i<=n;++i)
for(int re j=1;j<=m+1;++j)s[i][j]=add(s[i-1][j-1],mul(s[i-1][j],i-1));
for(int re i=1;i<=m;++i)
for(int re j=1;j<=m;++j)S[i][j]=add(S[i-1][j-1],mul(S[i-1][j],j));
}
inline int calc(int n,int m){
int res=0;
for(int re i=0;i<=m;++i)Inc(res,mul(S[m][i],mul(s[n+1][i+1],fac[i])));
return res;
}
public:
CyclesNumber(){}
std::vector<int> getExpectation(std::vector<int> N,std::vector<int> M){
init_stirling(*std::max_element(N.begin(),N.end())+1,
*std::max_element(M.begin(),M.end()));
std::vector<int> ans;ans.resize(N.size());
for(int re i=0;i<N.size();++i)ans[i]=calc(N[i],M[i]);
return ans;
}
};
#ifdef zxyoi
using std::cerr;
using std::cout;
CyclesNumber Solver;
inline void test1(){
freopen("test1.ans","w",stdout);
std::vector<int> ans=Solver.getExpectation({2},{2});
for(int re i:ans)cout<<i<<" ";
}
inline void test2(){
freopen("test2.ans","w",stdout);
std::vector<int> ans=Solver.getExpectation({3},{0});
for(int re i:ans)cout<<i<<" ";
}
inline void test3(){
freopen("test3.ans","w",stdout);
std::vector<int> ans=Solver.getExpectation({1,2,3},{1,3,3});
for(int re i:ans)cout<<i<<" ";
}
inline void test4(){
freopen("test4.ans","w",stdout);
std::vector<int> ans=Solver.getExpectation({10,20,30},{10,20,30});
for(int re i:ans)cout<<i<<" ";
}
signed main(){
test1();
test2();
test3();
test4();
return 0;
}
#endif