Solution:P5824(十二重计数法)

题目链接

Link:P5824

解题思路

$\text{An interesting problem.}$
我们把问题的要求这样分开：
满足球有标号的标上 $A$，满足盒子有标号的标上 $B$，满足至多放一个球的标 $C_1$，满足至少放一个球的标 $C_2$。$p=998244353$ 为模数。

Type $1:AB$

$\mathrm{I}:AB$

$\text{Rating:2 stars(Easy).}$
每个球都有 $m$ 个盒子可选，一共 $n$ 个球。
显然答案为 $m^n$。
时间复杂度 $\mathrm{O}(\log n)(\approx \mathrm{O}(1))$。

$\mathrm{I}\mathrm{I}:AB{C}_1$

$\text{Rating:2 stars(Easy).}$
$n>m$ 时，盒子装不下球，显然答案为 $0$。
否则答案显然为 $A_m^n$。
时间复杂度 $\mathrm{O}(\log p)(\approx \mathrm{O}(1))$。

$\mathrm{I}\mathrm{I}\mathrm{I}:AB{C}_2$

$\text{Rating:4 stars(Normal).}$
$n<m$ 时显然答案为 $0$。
否则，第二类Stirling数的意义就是把不同的球放到同样的盒子里。现在盒子不一样，全排列不就行了？
答案为 $m!\left\{\begin{array}{c}n\\ m\end{array}\right\}$。（其实 $\mathrm{V}\mathrm{I}$ 就是先计算这玩意再乘上 $\frac{1}{m!}$）
时间复杂度 $\mathrm{O}(m\log nmp)(\approx \mathrm{O}(m))$。

Type $2:A$

$\mathrm{I}\mathrm{V}:A$

$\text{Rating:6 stars(Hard).}$
看起来这玩意很像第二类Stirling数（$\mathrm{V}\mathrm{I}$）。但是，又不是。少了一个集合非空的条件。
那么，我们能不能把它的公式改一改呢？我没试过。不过，我们可以用另外一种方法。
枚举有几个盒子有球。我们得到
$$\sum _{i=1}^{\min (m,n)}\left\{\begin{array}{c}n\\ i\end{array}\right\}$$
好的，第二类Stirling数（行）的板子即可。
时间复杂度 $\mathrm{O}(n\log np)(\approx \mathrm{O}(n\log n))$。

$\mathrm{V}:A{C}_1$

$\text{Easiest Subtask.}$
$\text{Rating:1 star(Easy).}$
显然装得下是 $1$，装不下是 $0$。
时间复杂度 $\mathrm{O}(1)$。

$\mathrm{V}\mathrm{I}:A{C}_2$

$\text{Rating:3 stars(Normal).}$
$n<m$ 时显然答案为 $0$。
否则答案显然为 $\left\{\begin{array}{c}n\\ m\end{array}\right\}$。
时间复杂度 $\mathrm{O}(m\log nmp)(\approx \mathrm{O}(m))$。

Type $3:B$

$\mathrm{V}\mathrm{I}\mathrm{I}:B$

$\text{Rating:4 stars(Normal).}$
我们考虑先让每个盒子放一个球（即设球数为 $n+m$），即转化成 $\mathrm{I}\mathrm{X}$。
时间复杂度 $\mathrm{O}(\log p)(\approx \mathrm{O}(1))$。

$\mathrm{V}\mathrm{I}\mathrm{I}\mathrm{I}:B{C}_1$

$\text{Rating:2 stars(Easy).}$
$n>m$ 时，盒子装不下球，显然答案为 $0$。
否则答案显然为 $C_m^n$。
时间复杂度 $\mathrm{O}(\log p)(\approx \mathrm{O}(1))$。

$\mathrm{I}\mathrm{X}:B{C}_2$

$\text{Rating:3 stars(Normal).}$
用插板法可得答案为 $C_{n-1}^{m-1}$。
时间复杂度 $\mathrm{O}(\log p)(\approx \mathrm{O}(1))$。

Type $4:\varnothing$

$\mathrm{X}:\varnothing$

$\text{Rating:8 stars(Harder).}$
我们要求的答案是把 $n$ 划分为 $m$ 个自然数的方案数。设它为 $p_{n,m}$，有 $p_{n,m}=p_{n-m,m}+p_{n,m-1}$。考虑优化，设多项式
$$F_i(x)=\sum _{j=0}^{\infty }{p}_{j,i}{x}^j$$
则有
$$F_i(x)=F_{i-1}(x)\sum _{j=0}^{\infty }{x}^{ji}=\frac{F_{i-1}(x)}{1-x^i}=\prod _{j=1}^i\frac{1}{1-x^j}$$
参考付公主的背包（P4389） 的 $\ln +\exp$ 法，可得
$$\prod _{j=1}^i\frac{1}{1-x^j}=\exp \sum _{j=1}^i(-\ln (1-x^j))=\exp \sum _{j=1}^i\sum _{k=1}^{\infty }\frac{x^{jk}}{k}$$
我们要求的即为
$$[x^n]F_m=[x^n]\exp \sum _{i=1}^m\sum _{j=1}^{\infty }\frac{x^{ij}}{j}$$

$\mathrm{X}\mathrm{I}:C_1$

$\text{Easiest Subtask.}$
$\text{Rating:1 star(Easy).}$
显然装得下是 $1$，装不下是 $0$。
时间复杂度 $\mathrm{O}(1)$。

$\mathrm{X}\mathrm{I}\mathrm{I}:C_2$

$\text{Rating:9 stars(Insane).}$
我们先放 $m-n$ 个球进盒子里，便可转化为 $\mathrm{X}$。
时间复杂度为 $\mathrm{O}(n\log n)$（可使用 $\mathrm{X}$ 的结果直接回答，真实复杂度为 $\mathrm{O}(1)$）。

代码实现

//
#ifndef Polynomial_Class
//这里是长为13KB的多项式全家桶，由于篇幅原因省略
#endif

using namespace std;
using namespace Polynomial;

array<long long,400001> facts,inverses;
polynomial<524288> poly_n_4,poly_m_4,poly_4,poly_exp,poly_ans;

int main(int argc,char* argv[],char* envp[])
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	facts[0]=1;
	for(int i=1;i<=400000;i++)
		facts[i]=facts[i-1]*i%prime;
	long long cntbox,cntball;
	cin>>cntball>>cntbox;
	cout<<fast_pow(cntbox,cntball,prime)<<'\n'<<(cntball>cntbox?0:facts[cntbox]*inverse(facts[cntbox-cntball],prime)%prime)<<'\n';
	long long ans_3=0;
	if(cntball<cntbox)
	{
		cout<<0<<'\n';
		goto endcase_3;
	}
	for(int i=0;i<=cntbox;i++)
		ans_3=(ans_3+fast_pow(prime-1,i,prime)*facts[cntbox]%prime*inverse(facts[i]*facts[cntbox-i]%prime,prime)%prime*fast_pow(cntbox-i,cntball,prime)%prime)%prime;
	cout<<ans_3<<'\n';
	endcase_3:;
	long long ans_4=0;
	for(int i=0;i<=cntball;i++)
		poly_n_4[i]=fast_pow(i,cntball,prime)*inverse(facts[i],prime)%prime,poly_m_4[i]=fast_pow(prime-1,i,prime)*inverse(facts[i],prime)%prime;
	mult(cntball+1,cntball+1,poly_n_4,poly_m_4,poly_4);
	for(int i=1;i<=min(cntball,cntbox);i++)
		ans_4=(ans_4+poly_4[i])%prime;
	cout<<ans_4<<'\n'<<(cntball<=cntbox)<<'\n'<<ans_3*inverse(facts[cntbox],prime)%prime<<'\n'
		<<facts[cntball+cntbox-1]%prime*inverse(facts[cntbox-1]*facts[cntball]%prime,prime)%prime<<'\n'
		<<(cntbox<cntball?0:facts[cntbox]%prime*inverse(facts[cntball]*facts[cntbox-cntball]%prime,prime)%prime)<<'\n'
		<<(cntbox>cntball?0:facts[cntball-1]%prime*inverse(facts[cntbox-1]*facts[cntball-cntbox]%prime,prime)%prime)<<'\n';
	for(int i=1;i<=cntball;i++)
		inverses[i]=inverse(i,prime);
	for(int i=1;i<=min(cntbox,cntball);i++)
		for(int j=1;j*i<=cntball;j++)
			poly_exp[i*j]=(poly_exp[i*j]+inverses[j])%prime;
	exp(cntball+1,poly_exp,poly_ans);
	cout<<poly_ans[cntball]<<'\n'<<(cntball<=cntbox)<<'\n'<<(cntball<cntbox?0:poly_ans[cntball-cntbox]);
	return 0;
}