結果
問題 |
No.1741 Arrays and XOR Procedure
|
ユーザー |
![]() |
提出日時 | 2021-11-12 22:42:31 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 108 ms / 2,000 ms |
コード長 | 748 bytes |
コンパイル時間 | 410 ms |
コンパイル使用メモリ | 82,024 KB |
実行使用メモリ | 107,104 KB |
最終ジャッジ日時 | 2024-11-25 20:23:43 |
合計ジャッジ時間 | 4,902 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
""" A1+A3 , A2+A4 , A3+A """ import sys from sys import stdin def nCrB(n,r): binum = bi[n] - bi[r] - bi[n-r] if binum == 0: return 1 else: return 0 mod = 998244353 N = int(stdin.readline()) bi = [0] * (N+1) for i in range(1,30): x = 2**i for j in range(x,N+1,x): bi[j] += 1 for i in range(N): bi[i+1] += bi[i] B = list(map(int,stdin.readline().split())) C = [nCrB(N-1 , i) for i in range(N)] dp = [1,0] for i in range(N): ndp = [0,0] if B[i] == 0 or B[i] == -1: ndp[0] += dp[0] ndp[1] += dp[1] if B[i] == 1 or B[i] == -1: ndp[0 ^ C[i]] += dp[0] ndp[1 ^ C[i]] += dp[1] ndp[0] %= mod ndp[1] %= mod dp = ndp print (dp[1] % mod)