結果
問題 | No.1307 Rotate and Accumulate |
ユーザー | shotoyoo |
提出日時 | 2020-12-04 21:08:08 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 455 ms / 5,000 ms |
コード長 | 1,456 bytes |
コンパイル時間 | 159 ms |
コンパイル使用メモリ | 10,844 KB |
実行使用メモリ | 74,072 KB |
最終ジャッジ日時 | 2023-10-13 10:49:45 |
合計ジャッジ時間 | 8,912 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 124 ms
29,788 KB |
testcase_01 | AC | 117 ms
29,876 KB |
testcase_02 | AC | 120 ms
29,848 KB |
testcase_03 | AC | 129 ms
29,896 KB |
testcase_04 | AC | 129 ms
29,808 KB |
testcase_05 | AC | 129 ms
29,848 KB |
testcase_06 | AC | 126 ms
29,860 KB |
testcase_07 | AC | 128 ms
29,900 KB |
testcase_08 | AC | 424 ms
64,288 KB |
testcase_09 | AC | 383 ms
65,896 KB |
testcase_10 | AC | 317 ms
54,992 KB |
testcase_11 | AC | 261 ms
51,400 KB |
testcase_12 | AC | 299 ms
54,724 KB |
testcase_13 | AC | 164 ms
39,396 KB |
testcase_14 | AC | 201 ms
44,064 KB |
testcase_15 | AC | 439 ms
73,328 KB |
testcase_16 | AC | 449 ms
73,280 KB |
testcase_17 | AC | 437 ms
74,068 KB |
testcase_18 | AC | 413 ms
66,984 KB |
testcase_19 | AC | 437 ms
74,072 KB |
testcase_20 | AC | 455 ms
73,428 KB |
testcase_21 | AC | 119 ms
29,896 KB |
ソースコード
import sys input = lambda : sys.stdin.readline().rstrip() sys.setrecursionlimit(max(1000, 10**9)) write = lambda x: sys.stdout.write(x+"\n") import numpy as np # FFT import numpy as np M = 998244353 def fft(a,b): l = 1 while 2 * l < len(a) + len(b) - 1: l *= 2 l *= 2 c = np.fft.irfft((np.fft.rfft(a,l))*(np.fft.rfft(b,l)),l) c = np.rint(c).astype(np.int64) return c def fft_large(a,b): d = 30000 a1, a2 = np.divmod(a,d) b1, b2 = np.divmod(b,d) aa = fft(a1,b1) % M bb = fft(a2,b2) % M cc = (fft(a1+a2, b1+b2) - (aa+bb)) % M h = (((aa*d)%M)*d + cc*d + bb) % M return h def fft_large(a,b): """精度が足りないときはこちら """ d = 1<<10 a1, a2 = np.divmod(a,d*d) a2, a3 = np.divmod(a2,d) b1, b2 = np.divmod(b,d*d) b2, b3 = np.divmod(b2,d) aa = fft(a1,b1) % M bb = fft(a2,b2) % M cc = fft(a3,b3) % M dd = (fft(a1+a2, b1+b2) - (aa+bb)) % M ee = (fft(a2+a3, b2+b3) - (bb+cc)) % M ff = (fft(a1+a3, b1+b3) - (aa+cc)) % M h = (((aa*d*d)%M)*d*d + ((dd*d*d)%M)*d + (bb+ff)*d*d + ee*d + cc) % M return h n,q = list(map(int, input().split())) a = np.array(list(map(int, input().split())), dtype=np.int64) # r = np.array(list(map(int, input().split())), dtype=np.int64) r = list(map(int, input().split())) c = [0]*n for num in r: c[(-num)%n] += 1 v = fft_large(a,c) vv = (v[:n] + v[n:2*n]).tolist() write(" ".join(map(str, vv)))