CF1342E. Placing Rooks(容斥+组合数)

原题链接

题意:n * n的棋盘有n个车,满足下面的条件

1.棋盘上的每一个空格子至少能够被车走直线走到

2.恰好存在k辆车互相攻击

解法：

1.对于第一个条件来说,我们只需要进行考虑行的,对于列的*2即可,对于k=0不考虑。

2.首先我们每一个行都能够到了,所以确定了第一个条件,之后我们对于一个列放x对车,那么互相攻击就有x-1对了

3.分析完1,2后,我们只需要把点放在n-k列中进行排列组合,对于放在n-k列中,答案总共有(n-k)^n,之后我们需要减去多余情况,利用容斥原理,之后我们减去多余空行为1的,加上空行为2的,…之后就可以得到答案了。

#include<bits/stdc++.h>
#define ll long long 
#define pii pair<int,int>
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
#define x first
#define y second
using namespace std;
const int maxn = 2e5 + 5;
const int mod = 998244353;
ll fact[maxn],inv[maxn];
ll qpow(ll a, ll b, ll mod){
    ll ans = 1;
    while(b){
        if(b & 1)ans=1ll*ans*a%mod;
        b>>=1;
        a=1ll*a*a%mod;
    }
    return ans;
}
void init(){
    fact[0] = 1;
  for(int i = 1; i < maxn; i++)
    fact[i] = 1ll * fact[i - 1] * i % mod;
  inv[maxn - 1] = qpow(fact[maxn - 1], mod - 2, mod);
  for(int i = maxn - 2; i >= 0; i--)
    inv[i] = 1ll * inv[i + 1] * (i + 1) % mod;
}
ll C(ll n, ll m, ll mod){
    if(n < 0 || m < 0 || m > n)return 0;
    if(m == 0 || m == n)return 1;
    return 1ll*fact[n]*inv[m]%mod*inv[n - m]%mod;
}
int main(){
	IOS;
    init();
    ll n,k;
    cin >> n >> k;
    if(k == 0){
    	cout << fact[n]%mod << endl;
    	return 0;
	}
	if(k >= n){
		cout << "0" << endl;
		return 0;
	}
	//我们把每一个
	ll ans = 0;
	int f = 1;
	for(int i = 0; i <= n-k; i++){
		ans = (ans + f%mod*C(n-k,n-k-i,mod)%mod*qpow(n-k-i,n,mod)%mod + 2ll*mod)%mod;
		ans = (ans + mod)%mod;
//		cout << "ans = " << ans << endl;
		f = -f;
	} 
	ans = 2ll*C(n,n-k,mod)%mod*ans%mod;
	ans %= mod;
	cout << ans << endl;
}