Codeforces 1342E Placing Rooks(容斥+组合数学)

题目链接：E. Placing Rooks

题意：给定一个N*棋盘，要求摆放n个rooks（国际象棋里能横竖走的那啥）,使得每个格子都能被rooks攻击到，并且正好有k对rooks能相互攻击到,求有多少种摆放方式

题解：首先要求每个格子都能被访问，那么一定是每行都有一个或者每列都有一个，这两个的情况相当于翻转了棋盘，所有考虑每行都有一个的情况，最后答案*2即可（注意如果k=0，那么横竖情况正好重叠，此时不能*2），要使得恰好有k对能相互攻击，这里很玄妙，我们尝试几个例子能发现，有k对能相互攻击时，整个棋盘一定正好有k列是空着的（我也说不出来为啥，但是试了下确实很玄妙），那么题目就变成了在n-k列中摆放棋子，使得每列至少有一个的方案数，这就是经典容斥了，首先求总方案数(n-k)^n*C(n,n-k)，减去有一列为空时其他列自由组合的情况                     (n-k)^n*C(n-k-1,1)*C(n,n-k),再加上有两列为空时其他列自由组合的情况,......

具体细节和不懂看代码及注释（讲解部分不是很善于表达，但代码注释部分写的很详细）

#include<iostream>
#include<stack>
#include<list>
#include<set>
#include<vector>
#include<algorithm>
#include<math.h>
#include<numeric>
#include<map>
#include<cstring>
#include<queue>
#include<iomanip>
#include<cmath>
#include<queue>
#include <bitset>
#include<unordered_map>
	#ifndef local
	#define endl '\n'
#endif */
#define mkp make_pair
using namespace std;
using std::bitset;
typedef long long ll;
typedef long double ld;
const int inf=0x3f3f3f3f;
const ll MAXN=2e5+10;
const ll N=1e5+100;
const ll mod=998244353;
const ll hash_p1=1610612741;
const ll hash_p2=805306457;
const ll hash_p3=402653189;
//-----------------------------------------------------------------------------------------------------------------*/
// ll head[MAXN],net[MAXN],to[MAXN],edge[MAXN]/*流量*/,cost[MAXN]//费用;
/* 
void add(ll u,ll v,ll w,ll s){
	to[++cnt]=v;net[cnt]=head[u];edge[cnt]=w;cost[cnt]=s;head[u]=cnt;
	to[++cnt]=u;net[cnt]=head[v];edge[cnt]=0;cost[cnt]=-s;head[v]=cnt;
}
struct elemt{
	int p,v;
};
-----------------------------------
求[1,MAXN]组合式和逆元 
ll mi(ll a,ll b){
	ll res=1;
	while(b){
		if(b%2){
			res=res*a%mod;
		}	
		a=a*a%mod;
		b/=2;
	}
	return res;
}
ll fac[MAXN+10],inv[MAXN+10];
ll C(int m,int n){//组合式C(m,n); 
	if(!n){
		return 1;
	}
	return fac[m]*(inv[n]*inv[m-n]%mod)%mod;
}
void init(){
	fac[0]=1;inv[0]=1;
	for(ll i=1;i<=MAXN;i++){
		fac[i]=(fac[i-1]*i)%mod;
		inv[i]=mi(fac[i],mod-2);
	}
}
---------------------------------
 unordered_map<int,int>mp;
//优先队列默认小顶堆 ， greater<int> --小顶堆  less<int> --大顶堆  
priority_queue<elemt,vector<elemt>,comp>q;
struct comp{
	public:
		bool operator()(elemt v1,elemt v2){
			return v1.v<v2.v;
		}
};
	set<int>::iterator it=st.begin();
*/
// vector<vector<int>>edge; 二维虚拟储存坐标 
//-----------------------------------------------------------------------------------------------------------------*/
  //map<int,bool>mp[N]; 
ll mi(ll a,ll b){
	ll res=1;
	while(b){
		if(b%2){
			res=res*a%mod;
		}	
		a=a*a%mod;
		b/=2;
	}
	return res;
}
ll fac[MAXN+10],inv[MAXN+10];
ll C(int m,int n){//组合式C(m,n); 
	if(!n){
		return 1;
	}
	return fac[m]*(inv[n]*inv[m-n]%mod)%mod;
}
void init(){
	fac[0]=1;inv[0]=1;
	for(ll i=1;i<=MAXN;i++){
		fac[i]=(fac[i-1]*i)%mod;
		inv[i]=mi(fac[i],mod-2);
	}
}
int main(){
/*cout<<setiosflags(ios::fixed)<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（不含整数部分）*/
/*cout<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（含整数部分）*/
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);//同步流
	init();
	ll n,k;
	cin>>n>>k;
	if(k>n-1){//最多只能有n-1对
		cout<<0<<endl;
		return 0;
	}
	ll ans=mi(n-k,n)*C(n,n-k)%mod;//n个点在n-k个位置中的排列情况，
	for(int i=1;i<n-k;i++){//容斥
		if(i%2){//奇减
			ans-=(mi(n-k-i,n)*C(n-k,i)%mod)*C(n,n-k)%mod;//n-k列中有i列为空的情况
		}
		else{//偶加
			ans+=(mi(n-k-i,n)*C(n-k,i)%mod)*C(n,n-k)%mod;
		}
		ans=(ans+mod)%mod;
	}
	if(k){//当k为0时横竖的情况都一样就不用乘2了
		ans=ans*2ll%mod;
	}
	cout<<ans<<endl;
	return 0;
}