結果

問題 No.1741 Arrays and XOR Procedure
ユーザー wolgnikwolgnik
提出日時 2021-11-12 22:58:40
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 170 ms / 2,000 ms
コード長 661 bytes
コンパイル時間 697 ms
コンパイル使用メモリ 82,188 KB
実行使用メモリ 103,932 KB
最終ジャッジ日時 2024-11-25 20:59:04
合計ジャッジ時間 6,615 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 34 ms
53,892 KB
testcase_01 AC 34 ms
53,736 KB
testcase_02 AC 35 ms
53,224 KB
testcase_03 AC 157 ms
103,564 KB
testcase_04 AC 36 ms
52,016 KB
testcase_05 AC 158 ms
103,932 KB
testcase_06 AC 170 ms
103,396 KB
testcase_07 AC 160 ms
103,516 KB
testcase_08 AC 145 ms
103,332 KB
testcase_09 AC 143 ms
100,116 KB
testcase_10 AC 131 ms
100,304 KB
testcase_11 AC 132 ms
94,628 KB
testcase_12 AC 122 ms
89,876 KB
testcase_13 AC 134 ms
100,036 KB
testcase_14 AC 118 ms
91,972 KB
testcase_15 AC 135 ms
97,472 KB
testcase_16 AC 68 ms
75,752 KB
testcase_17 AC 83 ms
82,840 KB
testcase_18 AC 128 ms
92,364 KB
testcase_19 AC 67 ms
75,988 KB
testcase_20 AC 166 ms
103,928 KB
testcase_21 AC 136 ms
96,956 KB
testcase_22 AC 121 ms
94,280 KB
testcase_23 AC 91 ms
81,668 KB
testcase_24 AC 76 ms
76,056 KB
testcase_25 AC 139 ms
103,324 KB
testcase_26 AC 92 ms
85,152 KB
testcase_27 AC 58 ms
74,824 KB
testcase_28 AC 126 ms
94,612 KB
testcase_29 AC 122 ms
97,360 KB
testcase_30 AC 127 ms
94,604 KB
testcase_31 AC 64 ms
76,420 KB
testcase_32 AC 93 ms
85,336 KB
testcase_33 AC 100 ms
88,228 KB
testcase_34 AC 74 ms
79,280 KB
testcase_35 AC 42 ms
61,840 KB
testcase_36 AC 68 ms
76,308 KB
testcase_37 AC 99 ms
84,896 KB
testcase_38 AC 158 ms
103,348 KB
testcase_39 AC 83 ms
83,912 KB
testcase_40 AC 51 ms
65,836 KB
testcase_41 AC 69 ms
78,616 KB
testcase_42 AC 57 ms
73,340 KB
testcase_43 AC 35 ms
51,992 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
N = int(input())
a = list(map(int, input().split()))
mod = 998244353

def combi(n, k):
  if n < k: return 0
  if n == k: return 1
  if n > k: return 1

def lucas(n, k):
  res = 1
  while n and k:
    if n & 1 < k & 1: res = 0
    elif n & 1 == 0: res &= 1
    else: res &= combi(n & 1, k & 1)
    n >>= 1
    k >>= 1
  return res
dp = [0] * 2
dp[0] = 1
for i in range(N):
  c = lucas(N - 1, i)
  #print(c)
  if a[i] == 1:
    if c: dp[0], dp[1] = dp[1], dp[0]
  if a[i] == -1:
    if c == 0: dp[0], dp[1] = dp[0] * 2 % mod, dp[1] * 2 % mod
    else: dp[0], dp[1] = (dp[0] + dp[1]) % mod, (dp[0] + dp[1]) % mod
print(dp[-1])
0