计蒜客 - T3224

一、题意： 有n个黑球和m个白球，有n个空盒子，问如果黑球和白球不装在同一个盒子中，将所有的球都装进盒子里(可以有空盒子)的方案数有多少。答案需对998244353取模。
二、思路： 如果没有白球，就相当于将n个黑球装进n个盒子里(可空)的方案数，答案就是C(2n-1,n-1)
如果有白球，则可以从1到n-1遍历(因为至少要有一个盒子装白球，所以只能遍历到n-1)。
将n个黑球放入i个盒子(不空)的方案数为C(n-1,i-1)，在n个盒子中选i个盒子的方案数为C(n,i)，剩下的n-i个盒子中放白球(可空)的方案数为C(m + n - i - 1, n - i - 1)，所以总方案数为C(n-1,i-1)*C(n,i)*C(m + n - i - 1, n - i - 1)。
因为阶乘较大，所以需要预处理取余。在做除法取余的时候需要先求出逆元，因为mod为一个质数，所以求逆元的公式可以化简为
*val[i] = (mod - mod / i)val[mod%i] % mod(i为需要求逆元的数)
cpp代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<queue>
#include<vector>
#include<stack>
#define int long long 
using namespace std;
const int mod = 998244353;
int f[2000006];
int val[2000006];
void pre(int len) {
	f[0] = f[1]=1;
	for (int i = 1; i <= len; i++)
		f[i] = f[i - 1] * i%mod;
	val[0]=val[1] = 1;
	for (int i = 2; i <= len; i++)
		val[i] = (mod - mod / i)*val[mod%i] % mod;
	for (int i = 2; i <= len; i++)
		val[i] = val[i - 1] * val[i] % mod;
}
int c(int n, int m) {
	if (n < m)return 0;
	return f[n] * val[m] % mod*val[n - m] % mod;
}
signed main()
{
	int n, m;
	scanf("%lld%lld", &n, &m);
	if (m == 0) {
		pre(n + n);
		printf("%lld\n", c(n + n - 1, n - 1));
		return 0;
	}
	else {
		pre(n + m);
		int ans = 0;
		for (int i = 1; i < n; i++) {
			int tmp = c(n, i)*c(n - 1, i - 1) % mod*c(m + n - i - 1, n - i - 1) % mod;
			ans = (ans%mod) + (tmp%mod) % mod;
		}
		printf("%lld\n", ans%mod);
	}
	return 0;
}