結果

問題 No.1307 Rotate and Accumulate
ユーザー ayaoni
提出日時 2020-12-16 03:31:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 248 ms / 5,000 ms
コード長 2,076 bytes
コンパイル時間 281 ms
コンパイル使用メモリ 82,672 KB
実行使用メモリ 114,048 KB
最終ジャッジ日時 2024-09-20 04:25:30
合計ジャッジ時間 5,250 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

import sys
sys.setrecursionlimit(10**7)
def I(): return int(sys.stdin.readline().rstrip())
def MI(): return map(int,sys.stdin.readline().rstrip().split())
def LI(): return list(map(int,sys.stdin.readline().rstrip().split()))
def LI2(): return list(map(int,sys.stdin.readline().rstrip()))
def S(): return sys.stdin.readline().rstrip()
def LS(): return list(sys.stdin.readline().rstrip().split())
def LS2(): return list(sys.stdin.readline().rstrip())
# 998244353 = 119*2**23+1
mod = 998244353
primitive_root = 3 # mod
roots = [pow(primitive_root,(mod-1) >> i,mod) for i in range(24)]
inv_roots = [pow(r,mod-2,mod) for r in roots]
# roots[i] = 1 2**i inv_roots[i] = 1 2**i
#
def ntt(A,n):
for i in range(n):
m = 1 << (n-i-1)
for start in range(1 << i):
w = 1
start *= m*2
for j in range(m):
A[start+j],A[start+j+m] = (A[start+j]+A[start+j+m]) % mod,(A[start+j]-A[start+j+m])*w % mod
w *= roots[n-i]
w %= mod
return A
def inv_ntt(A,n):
for i in range(n):
m = 1 << i
for start in range(1 << (n-i-1)):
w = 1
start *= m*2
for j in range(m):
A[start+j],A[start+j+m] = (A[start+j]+A[start+j+m]*w) % mod,(A[start+j]-A[start+j+m]*w) % mod
w *= inv_roots[i+1]
w %= mod
a = pow(2,n*(mod-2),mod)
for i in range(1 << n):
A[i] *= a
A[i] %= mod
return A
def convolution(A,B):
a,b = len(A),len(B)
deg = a+b-2
n = deg.bit_length()
N = 1 << n
A += [0]*(N-a) # A 2-1
B += [0]*(N-b) # B 2-1
A = ntt(A,n)
B = ntt(B,n)
C = [(A[i]*B[i]) % mod for i in range(N)]
C = inv_ntt(C,n)
return C[:deg+1]
N,Q = MI()
A = LI()
R = LI()
C = [0]*(N+1)
for r in R:
C[N-r] += 1
B = convolution(A,C)
ANS = [0]*N
for i in range(len(B)):
ANS[i % N] += B[i]
ANS[i % N] %= mod
print(*ANS)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0