CF1342E Placing Rooks

最新推荐文章于 2021-08-16 21:24:20 发布

C202044zxy

最新推荐文章于 2021-08-16 21:24:20 发布

阅读量213

点赞数

分类专栏：容斥原理

本文链接：https://blog.csdn.net/C202044zxy/article/details/108462388

版权

容斥原理专栏收录该内容

11 篇文章 0 订阅

订阅专栏

一、题目

点此看题

二、解法

先考虑第一个条件，所有空格子被攻击到的充要条件是每一行或者每一列都要有车，由于行和列是完全不同的两种情况，所以可以只算行的情况再 $\times2$ （但是要特判 $k = 0$ 的情况，就不需要乘2）

在考虑如果得到 $k$ 个冲突，这等价于恰好 $k$ 个列没有车，那么原本在这些列上的车就会跑到其他的列形成冲突。这个问题就很想容斥了，原理大致是：我们先选出 $n - k$ 列恰好有车，然后用钦定 $0$ 列没有车的方案（在这 $n - k$ 列中）-钦定 $1$ 列+钦定 $2$ 列 $. . . . . .$ ，钦定的方案数是 $C(n-k,i)\times(n-k-i)^n$

#include <cstdio>
const int M = 200005;
const int MOD = 998244353;
#define int long long
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,m,ans,fac[M],inv[M];
void init(int n)
{
	inv[0]=inv[1]=fac[0]=fac[1]=1;
	for(int i=2;i<=n;i++) inv[i]=(MOD-MOD/i)*inv[MOD%i]%MOD;
	for(int i=2;i<=n;i++) inv[i]=inv[i]*inv[i-1]%MOD;
	for(int i=2;i<=n;i++) fac[i]=fac[i-1]*i%MOD;
}
int C(int n,int m)
{
	return fac[n]*inv[m]%MOD*inv[n-m]%MOD;
}
int qkpow(int a,int b)
{
	int r=1;
	while(b>0)
	{
		if(b&1) r=r*a%MOD;
		a=a*a%MOD;
		b>>=1;
	}
	return r;
}
signed main()
{
	init(200000);
	n=read();m=read();
	if(m>=n)
	{
		puts("0");
		return 0;
	}
	if(!m)
	{
		printf("%lld\n",fac[n]);
		return 0;
	}
	for(int i=0;i<=n-m;i++)
	{
		int tmp=C(n-m,i)*qkpow(n-m-i,n)%MOD;
		if(i&1) ans=(ans-tmp+MOD)%MOD;
		else ans=(ans+tmp)%MOD;
	}
	ans=2*ans*C(n,m)%MOD;
	printf("%lld\n",ans);
}
/*
\sum_{i=0}^{n-k} (-1)^i*C(n-k,i)*(n-k-i)^n
*/