結果

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

ソースコード

diff # 

P = 998244353
K = 216
KK = 1 << K
N, Q = map(int, input().split())
A = [1] * N + [KK + (int(a) - 1) % P for a in input().split()]
m1 = int(("1" * 72 + "0" * 144) * 200200, 2)
m2 = int(("1" * 120 + "0" * 96) * 200200, 2)
m3 = int(("1" * 152 + "0" * 64) * 200200, 2)
pa1 = (1 << 144) - ((1 << 144) % P)
pa2 = (1 << 96) - ((1 << 96) % P)
pa3 = (1 << 64) - ((1 << 64) % P)
def modP(x):
    x -= ((x & m1) >> 144) * pa1
    x -= ((x & m2) >> 96) * pa2
    x -= ((x & m3) >> 64) * pa3
    return x

for i in range(N)[::-1]:
    A[i] = modP(A[2*i] * A[2*i+1])

t = bin(A[1])[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])
s = 3 ** 120
t = modP(s)
0