結果

問題 No.1741 Arrays and XOR Procedure
コンテスト
ユーザー ygd.
提出日時 2021-11-13 11:48:56
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,357 bytes
コンパイル時間 408 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 34,940 KB
最終ジャッジ日時 2024-11-27 05:00:43
合計ジャッジ時間 119,208 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 2 TLE * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
#input = sys.stdin.readline
input = sys.stdin.buffer.readline

def cmb(n,k):
    if (k<0) or (n<k): return 0
    ret = 1
    for i in range(2,n+1):
        ret *= i
    for i in range(2,k+1):
        ret //= i
    for i in range(2,(n-k)+1):
        ret //= i
    return ret

def lucas(n,k,p):
    keta = 1
    m = n
    while m > 0:
        keta += 1
        m //= p

    ret = 1    
    for i in range(keta):
        nn = n>>i
        kk = k>>i
        temp = cmb(nn,kk)%p
        ret = ret*temp%p
    return ret
        


def main():
    n = int(input()); MOD = 998244353
    B = list(map(int,input().split()))

    #dp[i][j]:i番目まで見て和が0(偶数)/1(奇数)
    dp = [1,0]
    for i in range(n):
        p = [0,0]
        p,dp = dp,p
        pari = lucas(n-1,i,2)
        if B[i] == 0:
            dp[0] = p[0]
            dp[1] = p[1]
        elif B[i] == 1:
            if pari == 0:
                dp[0] = p[0]
                dp[1] = p[1]
            else:
                dp[0] = p[1]
                dp[1] = p[0]
        else:
            if pari == 0:
                dp[0] = 2*p[0]
                dp[1] = 2*p[1]
            else:
                dp[0] = p[0] + p[1]
                dp[1] = p[0] + p[1]
        dp[0] %= MOD
        dp[1] %= MOD

    ans = dp[1]
    print(ans%MOD)



if __name__ == '__main__':
    main()
0