【Atcoder】 [AGC040C] Neither AB nor BA

最新推荐文章于 2024-09-15 08:16:50 发布

Farmer_D

最新推荐文章于 2024-09-15 08:16:50 发布

阅读量119

点赞数

分类专栏： Atcoder 文章标签：算法

本文链接：https://blog.csdn.net/djx2021/article/details/132626176

版权

Atcoder 专栏收录该内容

29 篇文章 0 订阅

订阅专栏

题目链接

Atcoder方向
 Luogu方向

题目解法

考虑一步神仙的转化
把原图黑白染色，每次一定只会删除黑点和白点
所以如果把黑点上的 $A$ 变成 $B$ ， $B$ 变成 $A$ ，那么限制就等价于不能删 $AA, BB$
显然，一个序列不合法，当且仅当 $A$ 或 $B$ 为绝对众数
于是便可以容斥了
时间复杂度 $O (n)$

#include <bits/stdc++.h>
using namespace std;
const int N=10000100,P=998244353;
int n,pw[N],fac[N],inv[N];
inline int read(){
	int FF=0,RR=1;
	char ch=getchar();
	for(;!isdigit(ch);ch=getchar()) if(ch=='-') RR=-1;
	for(;isdigit(ch);ch=getchar()) FF=(FF<<1)+(FF<<3)+ch-48;
	return FF*RR;
}
int qmi(int a,int b){
	int res=1;
	for(;b;b>>=1){
		if(b&1) res=1ll*res*a%P;
		a=1ll*a*a%P;
	}
	return res;
}
int C(int a,int b){ return 1ll*fac[a]*inv[b]%P*inv[a-b]%P;}
int main(){
	n=read();
	fac[0]=1;
	for(int i=1;i<=n;i++) fac[i]=1ll*fac[i-1]*i%P;
	inv[n]=qmi(fac[n],P-2);
	for(int i=n-1;i>=0;i--) inv[i]=1ll*inv[i+1]*(i+1)%P;
	pw[0]=1;
	for(int i=1;i<=n;i++) pw[i]=pw[i-1]*2%P;
	//start 容斥
	int ans=qmi(3,n);
	for(int i=n/2+1;i<=n;i++){
		int res=1ll*C(n,i)*pw[n-i]%P;
        ans=(ans-res*2%P+P)%P;
	}
	printf("%d",ans);
	return 0;
}