結果

問題 No.1068 #いろいろな色 / Red and Blue and more various colors (Hard)
コンテスト
ユーザー Kiri8128
提出日時 2020-05-30 01:58:00
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 450 bytes
コンパイル時間 386 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 41,688 KB
最終ジャッジ日時 2024-11-06 15:08:12
合計ジャッジ時間 8,221 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 1 -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, Q = map(int, input().split())
P = 998244353
K = 96
A = [(int(a) - 1) % P for a in input().split()]
m = int(("1" * 32 + "0" * 64) * 5010, 2)
pa = (1 << 64) - ((1 << 64) % P)
modP = lambda x: x - ((x & m) >> 64) * pa
s = 1
for i, a in enumerate(A):
    s *= (1 << K) + a
    s = modP(s)

t = bin(s)[2:] + "_"
X = [int(t[-(i+1) * K - 1:-i * K - 1], 2) % P for i in range((len(t)+K-2) // K)]
for a in [int(a) for a in input().split()]:
    print(X[a])
0