結果

問題 No.2149 Vanitas Vanitatum
ユーザー 👑 rin204rin204
提出日時 2022-12-09 14:35:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 71 ms / 2,000 ms
コード長 1,345 bytes
コンパイル時間 329 ms
コンパイル使用メモリ 82,112 KB
実行使用メモリ 65,024 KB
最終ジャッジ日時 2024-04-22 19:10:49
合計ジャッジ時間 2,377 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
51,968 KB
testcase_01 AC 36 ms
52,224 KB
testcase_02 AC 44 ms
60,032 KB
testcase_03 AC 36 ms
51,840 KB
testcase_04 AC 36 ms
52,096 KB
testcase_05 AC 36 ms
52,480 KB
testcase_06 AC 35 ms
52,352 KB
testcase_07 AC 35 ms
52,480 KB
testcase_08 AC 36 ms
52,224 KB
testcase_09 AC 36 ms
52,096 KB
testcase_10 AC 37 ms
52,224 KB
testcase_11 AC 36 ms
52,096 KB
testcase_12 AC 38 ms
52,480 KB
testcase_13 AC 40 ms
52,480 KB
testcase_14 AC 38 ms
53,376 KB
testcase_15 AC 37 ms
52,884 KB
testcase_16 AC 37 ms
51,840 KB
testcase_17 AC 40 ms
52,736 KB
testcase_18 AC 37 ms
52,736 KB
testcase_19 AC 61 ms
63,872 KB
testcase_20 AC 65 ms
64,896 KB
testcase_21 AC 51 ms
62,080 KB
testcase_22 AC 65 ms
65,024 KB
testcase_23 AC 61 ms
63,488 KB
testcase_24 AC 36 ms
52,608 KB
testcase_25 AC 56 ms
62,848 KB
testcase_26 AC 71 ms
63,744 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

def HookLengthFormula(A, MOD=998244353):
    n = len(A)
    if n == 0:
        return 1
    tot = sum(A)
    ans = 1
    for i in range(1, tot + 1):
        ans *= i
        ans %= MOD
    
    inv = 1
    r = n - 1
    for i in range(A[0]):
        while A[r] == i:
            r -= 1
        for j in range(r, -1, -1):
            h = r - j + A[j] - i
            inv *= h
            inv %= MOD
    ans = ans * pow(inv, MOD - 2, MOD) % MOD
    return ans

n = int(input())
A = list(map(int, input().split()))
tot = sum(A)
if tot & 1:
    print(0)
    exit()

binary = []
b = 0
for a in A:
    binary += [0] * (a - b) + [1]
    b = a

P = binary[0::2]
Q = binary[1::2]

op = P.count(1)
oq = Q.count(1)
if op == oq:
    pass
elif len(P) == len(Q) and op + 1 == oq:
    pass
elif op == oq + 1:
    pass
else:
    print(0)
    exit()

def f(P):
    A = []
    x = 0
    for p in P:
        if p == 0:
            x += 1
        else:
            A.append(x)

    A = A[::-1]
    
    return HookLengthFormula(A), sum(A)

def nCk(n, k):
    ret = 1
    for i in range(n, n - k, -1):
        ret *= i
        ret %= MOD
    inv = 1
    for i in range(1, k + 1):
        inv *= i
        inv %= MOD
    return ret * pow(inv, MOD - 2, MOD) % MOD

pp, lp = f(P)
qq, lq  = f(Q)
ans = (pp * qq % MOD) * nCk(lp + lq, lp) % MOD
print(ans)
0