正题
题目链接:https://www.luogu.com.cn/problem/AT5661
题目大意
一个包含 A , B , C A,B,C A,B,C的序列,每次可以选择相邻的两个除了 A B AB AB和 B A BA BA的删去。
求有多少个长度为 N N N的序列可以删完。
1 ≤ N ≤ 1 0 7 1\leq N\leq 10^7 1≤N≤107
解题思路
因为每次是删除一个奇数位置和一个偶数位置,如果我们把所有偶数位置的取反,那么就变成了不能删除 A A AA AA和 B B BB BB。
然后如果在边上 A A A一定可以删(除非到边界),也就是 A A A的数量不能超过 n 2 \frac{n}{2} 2n,同理 B B B也是。
然后减去 A A A大于的或者 B B B大于的就好了(因为只能有一个大于)
时间复杂度 O ( n ) O(n) O(n)
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=1e7+10,P=998244353;
ll n,pw[N],inv[N],fac[N],ans;
ll C(ll n,ll m)
{return fac[n]*inv[m]%P*inv[n-m]%P;}
signed main()
{
scanf("%lld",&n);
pw[0]=ans=inv[1]=1;
for(ll i=2;i<=n;i++)inv[i]=P-(P/i)*inv[P%i]%P;
inv[0]=fac[0]=1;
for(ll i=1;i<=n;i++)
fac[i]=fac[i-1]*i%P,inv[i]=inv[i-1]*inv[i]%P;
for(ll i=1;i<=n;i++)
ans=ans*3%P,pw[i]=pw[i-1]*2%P;
for(ll i=n/2+1;i<=n;i++)
ans=(ans-pw[n-i]*C(n,i)%P*2%P+P)%P;
printf("%lld\n",ans);
return 0;
}