結果
問題 | No.1618 Convolution? |
ユーザー | shotoyoo |
提出日時 | 2021-07-22 22:59:20 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,004 ms / 2,000 ms |
コード長 | 1,456 bytes |
コンパイル時間 | 373 ms |
コンパイル使用メモリ | 10,880 KB |
実行使用メモリ | 130,624 KB |
最終ジャッジ日時 | 2023-09-24 18:47:01 |
合計ジャッジ時間 | 18,091 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 133 ms
29,800 KB |
testcase_01 | AC | 132 ms
29,796 KB |
testcase_02 | AC | 851 ms
113,216 KB |
testcase_03 | AC | 910 ms
121,668 KB |
testcase_04 | AC | 576 ms
86,384 KB |
testcase_05 | AC | 175 ms
35,492 KB |
testcase_06 | AC | 917 ms
118,308 KB |
testcase_07 | AC | 942 ms
119,660 KB |
testcase_08 | AC | 573 ms
86,660 KB |
testcase_09 | AC | 978 ms
128,308 KB |
testcase_10 | AC | 881 ms
109,960 KB |
testcase_11 | AC | 981 ms
125,776 KB |
testcase_12 | AC | 992 ms
129,564 KB |
testcase_13 | AC | 1,002 ms
129,232 KB |
testcase_14 | AC | 997 ms
130,560 KB |
testcase_15 | AC | 1,004 ms
129,228 KB |
testcase_16 | AC | 988 ms
130,624 KB |
ソースコード
import sys input = lambda : sys.stdin.readline().rstrip() sys.setrecursionlimit(2*10**5+10) write = lambda x: sys.stdout.write(x+"\n") debug = lambda x: sys.stderr.write(x+"\n") writef = lambda x: print("{:.12f}".format(x)) # FFT import numpy as np TYPE = np.int64 def fft(a,b): l = len(a) + len(b) - 1 l = 1<<((l-1).bit_length()) c = np.fft.irfft((np.fft.rfft(a,l))*(np.fft.rfft(b,l)),l) c = np.rint(c).astype(TYPE) 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) bb = fft(a2,b2) cc = fft(a3,b3) dd = (fft(a1+a2, b1+b2) - (aa+bb)) ee = (fft(a2+a3, b2+b3) - (bb+cc)) ff = (fft(a1+a3, b1+b3) - (aa+cc)) h = (((aa*d*d))*d*d + ((dd*d*d))*d + (bb+ff)*d*d + ee*d + cc) return h n = int(input()) a = list(map(int, input().split())) b = list(map(int, input().split())) a = np.array(a, dtype=np.int64) b = np.array(b, dtype=np.int64) aa = fft_large(np.arange(1,n+1).astype(np.int64), a) bb = fft_large(np.arange(1,n+1), b) ans = [0] + (aa+bb)[:2*n-1].tolist() write(" ".join(map(str, ans)))