Placing Rooks(2300/容斥/组合)

题目
官方&代码
题意：给定$n\ast n$棋盘，放置$n$个骑士，要求：1、每个空位置都要与至少一位骑士同一行或同一列。2、刚好有$k$对骑士产生行或列的碰撞。一个骑士，可以通过走他所在行，或者所在列，走到另一个骑士，中间无其他骑士障碍，那么这2个骑士属于一对碰撞。
思路：由于每个位置都要给骑士访问到，那么这$n$个骑士要刚好占据$n$行，或者占据$n$列，这2个情况等效，我们只需讨论其一，最后答案再$\ast 2$即可，但注意当$k=0$时不需$\ast 2$。
$n$个骑士占据了$n$行，要产生$k$对碰撞，画图看看，容易发现，要求刚好有$n-k$个空列，即$n$个骑士在$k$列中。这是个排列组合问题。
从$n$列中选取出$c=n-k$列出来，对于这$c$列，共有$c^n$种选择，但要减去有一个空列的情况$(c-1{)}^n\ast C_c^1$，加上有2个空列的情况$(c-2{)}^n\ast C_c^2$…，容斥下去，答案为${\sum }_{i=0}^c(-1{)}^i\ast (c-i{)}^n\ast C_c^i$
代码源自官方题解

#include<bits/stdc++.h>

using namespace std;

const int MOD = 998244353;
const int N = 200043;

int add(int x, int y)
{
	return (x + y) % MOD;
}

int sub(int x, int y)
{
	return add(x, MOD - y);
}

int mul(int x, int y)
{
	return (x * 1ll * y) % MOD;
}

int binpow(int x, int y)
{
	int z = 1;
	while(y > 0)
	{
		if(y % 2 == 1)
			z = mul(z, x);
		x = mul(x, x);
		y /= 2;
	}
	return z;
}

int inv(int x)
{
	return binpow(x, MOD - 2);
}

int fact[N];

int C(int n, int k)
{
	return mul(fact[n], inv(mul(fact[k], fact[n - k])));
}

int main()
{
	int n, k;
	cin >> n >> k;
	
	if(k > n - 1)
	{
		cout << 0 << endl;
		return 0;
	}
	
	fact[0] = 1;
	for(int i = 1; i <= n; i++)
		fact[i] = mul(i, fact[i - 1]);
	int ans = 0;
	int c = n - k;
	for(int i = c; i >= 0; i--)
		if(i % 2 == c % 2)
			ans = add(ans, mul(binpow(i, n), C(c, i)));
		else
			ans = sub(ans, mul(binpow(i, n), C(c, i)));
	ans = mul(ans, C(n, c));
	if(k > 0)
		ans = mul(ans, 2);
	cout << ans << endl;
}