結果
| 問題 | No.2506 Sum of Weighted Powers |
| ユーザー |
 suisen
|
| 提出日時 | 2023-03-22 00:00:38 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
WA
(最新)
AC
(最初)
|
| 実行時間 | - |
| コード長 | 6,243 bytes |
| コンパイル時間 | 240 ms |
| コンパイル使用メモリ | 82,048 KB |
| 実行使用メモリ | 164,348 KB |
| 最終ジャッジ日時 | 2024-09-22 23:44:41 |
| 合計ジャッジ時間 | 14,358 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge3 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 79 ms
67,968 KB |
| testcase_01 | WA | - |
| testcase_02 | AC | 74 ms
68,096 KB |
| testcase_03 | WA | - |
| testcase_04 | WA | - |
| testcase_05 | WA | - |
| testcase_06 | WA | - |
| testcase_07 | WA | - |
| testcase_08 | WA | - |
| testcase_09 | WA | - |
| testcase_10 | WA | - |
| testcase_11 | WA | - |
| testcase_12 | WA | - |
| testcase_13 | WA | - |
| testcase_14 | WA | - |
| testcase_15 | WA | - |
| testcase_16 | WA | - |
| testcase_17 | WA | - |
| testcase_18 | WA | - |
| testcase_19 | WA | - |
| testcase_20 | WA | - |
| testcase_21 | WA | - |
| testcase_22 | WA | - |
| testcase_23 | WA | - |
| testcase_24 | WA | - |
| testcase_25 | WA | - |
| testcase_26 | WA | - |
| testcase_27 | WA | - |
| testcase_28 | WA | - |
| testcase_29 | WA | - |
| testcase_30 | WA | - |
| testcase_31 | WA | - |
| testcase_32 | WA | - |
| testcase_33 | WA | - |
| testcase_34 | WA | - |
| testcase_35 | WA | - |
| testcase_36 | WA | - |
| testcase_37 | WA | - |
| testcase_38 | AC | 64 ms
68,224 KB |
| testcase_39 | AC | 64 ms
67,968 KB |
| testcase_40 | AC | 65 ms
68,352 KB |
| testcase_41 | WA | - |
ソースコード
from typing import List
def bsf(x):
res = 0
while not (x & 1):
res += 1
x >>= 1
return res
P = 998244353
G = 3
rank2 = bsf(P - 1)
class NTT:
class __RootInitializer:
@staticmethod
def root():
return [pow(G, (P - 1) >> i, P) for i in range(0, rank2 + 1)]
@staticmethod
def iroot():
return [pow(pow(G, P - 2, P), (P - 1) >> i, P) for i in range(0, rank2 + 1)]
root = __RootInitializer.root()
iroot = __RootInitializer.iroot()
class __RateInitializer:
@staticmethod
def rates(root: List[int], iroot: List[int]):
rate2 = [0] * max(0, rank2 - 1)
irate2 = [0] * max(0, rank2 - 1)
prod = iprod = 1
for i in range(rank2 - 1):
rate2[i] = root[i + 2] * prod % P
irate2[i] = iroot[i + 2] * iprod % P
prod = prod * iroot[i + 2] % P
iprod = iprod * root[i + 2] % P
rate3 = [0] * max(0, rank2 - 2)
irate3 = [0] * max(0, rank2 - 2)
prod = iprod = 1
for i in range(rank2 - 2):
rate3[i] = root[i + 3] * prod % P
irate3[i] = iroot[i + 3] * iprod % P
prod = prod * iroot[i + 3] % P
iprod = iprod * root[i + 3] % P
return rate2, irate2, rate3, irate3
rate2, irate2, rate3, irate3 = __RateInitializer.rates(__RootInitializer.root(), __RootInitializer.iroot())
@staticmethod
def butterfly(a: List[int]) -> None:
n = len(a)
h = bsf(n)
l = 0
while l < h:
if h - l == 1:
p = 1 << (h - l - 1)
rot = 1
for s in range(1 << l):
offset = s << (h - l)
for i in range(p):
u = a[i + offset]
v = a[i + offset + p] * rot
a[i + offset] = (u + v) % P
a[i + offset + p] = (u - v) % P
if s + 1 != 1 << l:
rot = rot * NTT.rate2[bsf(~s)] % P
l += 1
else:
p = 1 << (h - l - 2)
rot, imag = 1, NTT.root[2]
for s in range(1 << l):
rot2 = rot * rot % P
rot3 = rot2 * rot % P
offset = s << (h - l)
for i in range(p):
a0 = a[i + offset]
a1 = a[i + offset + p] * rot
a2 = a[i + offset + 2 * p] * rot2
a3 = a[i + offset + 3 * p] * rot3
a1na3imag = (a1 - a3) % P * imag
a[i + offset] = (a0 + a2 + a1 + a3) % P
a[i + offset + 1 * p] = (a0 + a2 - a1 - a3) % P
a[i + offset + 2 * p] = (a0 - a2 + a1na3imag) % P
a[i + offset + 3 * p] = (a0 - a2 - a1na3imag) % P
if s + 1 != (1 << l):
rot = rot * NTT.rate3[bsf(~s)] % P
l += 2
@staticmethod
def butterfly_inv(a : List[int]) -> None:
n = len(a)
h = bsf(n)
l = h
while l:
if l == 1:
p = 1 << (h - l)
irot = 1
for s in range(1 << (l - 1)):
offset = s << (h - l + 1)
for i in range(p):
u = a[i + offset]
v = a[i + offset + p]
a[i + offset] = (u + v) % P
a[i + offset + p] = ((u - v) * irot) % P
if s + 1 != 1 << (l - 1):
irot = irot * NTT.irate2[bsf(~s)] % P
l -= 1
else:
p = 1 << (h - l)
irot = 1
iimag = NTT.iroot[2]
for s in range(1 << (l - 2)):
irot2 = irot * irot % P
irot3 = irot2 * irot % P
offset = s << (h - l + 2)
for i in range(p):
a0 = a[i + offset]
a1 = a[i + offset + p]
a2 = a[i + offset + 2 * p]
a3 = a[i + offset + 3 * p]
a2na3iimag = (a2 - a3) * iimag % P
a[i + offset] = (a0 + a1 + a2 + a3) % P
a[i + offset + p] = ((a0 - a1 + a2na3iimag) * irot) % P
a[i + offset + 2 * p] = ((a0 + a1 - a2 - a3) * irot2) % P
a[i + offset + 3 * p] = ((a0 - a1 - a2na3iimag) * irot3) % P
if s + 1 != 1 << (l - 2):
irot = irot * NTT.irate3[bsf(~s)] % P
l -= 2
@staticmethod
def convolution(a, b):
n = len(a)
m = len(b)
if not a or not b:
return []
if min(n, m) <= 40:
if n < m:
n, m = m, n
a, b = b, a
res = [0] * (n + m - 1)
for i in range(n):
for j in range(m):
res[i + j] += a[i] * b[j]
res[i + j] %= P
return res
z = 1 << ((n + m - 1).bit_length())
iz = pow(z, P - 2, P)
a += [0] * (z - n)
b += [0] * (z - m)
NTT.butterfly(a)
NTT.butterfly(b)
c = [a[i] * b[i] % P * iz % P for i in range(z)]
NTT.butterfly_inv(c)
return c[:n + m - 1]
n, x = map(int, input().split())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
C = list(map(int, input().split()))
if x == 0:
ans = sum(A[i] * B[i] % P * C[0] % P for i in range(n + 1)) + sum(A[i] * B[0] % P * C[i] % P for i in range(1, n + 1))
print(ans % P)
else:
t = lambda k : (k - 1) * k * (k + 1) // 3
inv_x = pow(x, P - 2, P)
F = [B[i] * pow(inv_x, t(i), P) for i in range(n + 1)]
G = [C[i] * pow(inv_x, t(i), P) for i in range(n + 1)]
H = NTT.convolution(F, G)
ans = sum(A[i] * pow(x, t(i), P) % P * H[i] % P for i in range(n + 1))
print(ans % P)