【TC SRM 686 Div1】【TC 14199】CyclesNumber（斯特林数）

传送门

---

题解：

首先我们可以直接知道答案为$\sum s_{n,k}\cdot k^m$

用数学方法处理稍微麻烦了一点，考虑其组合意义。

设$x_i$表示循环$i$是否出现，则一个置换的权值为$(\sum x_i{)}^m$

展开，考虑其中由某$k$个项组成的项的系数，相当于是$m$个物品划分到$k$个带标号集合中的方案数，则系数为$S_{m,k}k!$。而这里$x_i$只可能为$0/1$，则系数就是对答案的最后贡献。

考虑某$k$个循环，如果它们同时出现，就会对答案有$S_{m,k}k!$的贡献。

那么考虑这$k$个以外剩下的循环，我们新加一个点$n+1$，剩下的循环按照字典顺序与$n+1$可以连成一个新的环，并且这样的方案是唯一的。所以选出前$k$个环的方案数为$s_{n+1,k+1}$。

则答案为${\sum }_k{s}_{n+1,k+1}{S}_{m,k}k!$

---

代码：

#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