@atcoder - AGC040C@ Neither AB nor BA

目录

• @description@
• @solution@
• @accepted code@
• @detail@

---

@description@

给定偶数 N，求由 'A', 'B', 'C' 三种字符组成的字符串 S，有多少知足以下的条件：

每次能够选择 S 中的两个相邻字符（不能选择 "AB" 与 "BA"），删除它们。最后能够将 S 删成空串。

好比："ABBC" -> "AC" -> ""。因此 "ABBC" 对于 N = 4 时是合法的。

将最终答案 mod 998244353。

Constraints
2≤N≤10^7, 并保证 N 为偶数

Input
输入形式以下：
N

Output
输出答案 mod 998244353。

Sample Input 1
2
Sample Output 1
7

除了 "AB", "BA" 均可行。

@solution@

考虑删除连续 2 个字符，哪些东西不会变化。
这时你会惊讶地发现：一个字符在字符串中的所处位置的奇偶性不会变化。
其实挺容易验证。假如在删除的前面，不会影响；假如在删除的后面，位置向前移动 2，奇偶不变。

那么一个奇数位置上的 "A" 与一个偶数位置上的 "B" 永远不可能互相消；一个偶数位置上的 "B" 与一个奇数位置上的 "A" 也永远不可能互相消。这些字符须要其余的字符消掉。
记 S1 = 奇数位置的 "A" 数量 + 偶数位置的 "B" 数量，那么应有 2*S1 <= N。
同理记 S2 = 奇数位置的 "B" 数量 + 偶数位置的 "A" 数量，那么应有 2*S2 <= N。

其实以上两个条件 (2*S1 <= N, 2*S2 <= N) 就是充要条件。
能够概括验证。假如 2*S1 = N, 2*S2 = N 同时知足，则咱们能够同时消 S1 与 S2；不然我老是消 S1 与 S2 的较大值。

那么最终答案就是 3^N - (2*S1 > N 的方案数) - (2*S2 > N 的方案数)。
随便组合计数一下就没了。

@accepted code@

#include <cstdio>
const int MOD = 998244353;
const int MAXN = 10000000;
int sub(int x, int y) {return x - y < 0 ? x - y + MOD : x - y;}
int pow_mod(int b, int p) {
    int ret = 1;
    while( p ) {
        if( p & 1 ) ret = 1LL*ret*b%MOD;
        b = 1LL*b*b%MOD;
        p >>= 1;
    }
    return ret;
}
int pw2[MAXN + 5], fct[MAXN + 5], ifct[MAXN + 5];
void init() {
    pw2[0] = 1;
    for(int i=1;i<=MAXN;i++)
        pw2[i] = 2LL*pw2[i-1]%MOD;
    fct[0] = 1;
    for(int i=1;i<=MAXN;i++)
        fct[i] = 1LL*fct[i-1]*i%MOD;
    ifct[MAXN] = pow_mod(fct[MAXN], MOD-2);
    for(int i=MAXN-1;i>=0;i--)
        ifct[i] = 1LL*ifct[i+1]*(i+1)%MOD;
}
int comb(int n, int m) {
    return 1LL*fct[n]*ifct[m]%MOD*ifct[n-m]%MOD;
}
int main() {
    init(); int N;
    scanf("%d", &N);
    int ans = pow_mod(3, N);
    for(int i=N/2+1;i<=N;i++)
        ans = sub(ans, 2LL*comb(N, i)*pw2[N-i]%MOD);
    printf("%d\n", ans);
}

@detail@

老年选手连 AGC 的 C 题都作不出来了 QAQ。
整了一个上午 + 一个晚上，最后仍是看了题解 QAQ。

感受主要是。。。想不到根据位置的奇偶性来分析吧。。。
吃一堑，长一智.jpg。

为何 AGC 这么喜欢出这种类型的题啊 QAQ。 人类智慧实在是太强大了 QAQ。