結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
52,224 KB
testcase_01 AC 43 ms
52,224 KB
testcase_02 AC 48 ms
59,776 KB
testcase_03 AC 40 ms
51,968 KB
testcase_04 AC 40 ms
52,096 KB
testcase_05 AC 40 ms
51,968 KB
testcase_06 AC 41 ms
52,224 KB
testcase_07 AC 40 ms
51,968 KB
testcase_08 AC 41 ms
52,096 KB
testcase_09 AC 41 ms
52,224 KB
testcase_10 AC 41 ms
52,096 KB
testcase_11 AC 41 ms
51,968 KB
testcase_12 AC 41 ms
52,224 KB
testcase_13 AC 40 ms
52,480 KB
testcase_14 AC 42 ms
52,992 KB
testcase_15 AC 41 ms
52,992 KB
testcase_16 AC 41 ms
52,224 KB
testcase_17 AC 41 ms
51,968 KB
testcase_18 AC 40 ms
52,480 KB
testcase_19 AC 64 ms
63,488 KB
testcase_20 AC 72 ms
64,896 KB
testcase_21 AC 57 ms
62,080 KB
testcase_22 AC 72 ms
64,512 KB
testcase_23 AC 68 ms
63,616 KB
testcase_24 AC 41 ms
52,096 KB
testcase_25 AC 63 ms
62,976 KB
testcase_26 AC 79 ms
63,616 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