結果

問題 No.2149 Vanitas Vanitatum
コンテスト
ユーザー rin204
提出日時 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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

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