結果
問題 | No.931 Multiplicative Convolution |
ユーザー | shotoyoo |
提出日時 | 2021-07-06 21:43:44 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,018 ms / 2,000 ms |
コード長 | 2,530 bytes |
コンパイル時間 | 96 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 78,860 KB |
最終ジャッジ日時 | 2024-07-01 12:15:51 |
合計ジャッジ時間 | 16,207 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 521 ms
44,124 KB |
testcase_01 | AC | 521 ms
44,252 KB |
testcase_02 | AC | 520 ms
44,384 KB |
testcase_03 | AC | 514 ms
43,996 KB |
testcase_04 | AC | 522 ms
44,380 KB |
testcase_05 | AC | 519 ms
44,764 KB |
testcase_06 | AC | 526 ms
44,128 KB |
testcase_07 | AC | 609 ms
46,176 KB |
testcase_08 | AC | 1,018 ms
78,860 KB |
testcase_09 | AC | 890 ms
67,364 KB |
testcase_10 | AC | 988 ms
77,732 KB |
testcase_11 | AC | 908 ms
66,692 KB |
testcase_12 | AC | 806 ms
64,356 KB |
testcase_13 | AC | 996 ms
77,984 KB |
testcase_14 | AC | 1,015 ms
78,464 KB |
testcase_15 | AC | 1,005 ms
78,268 KB |
testcase_16 | AC | 998 ms
77,716 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)) def factor(n, m=None): # mを与えると、高々その素因数まで見て、残りは分解せずにそのまま出力する f = {} tmp = n M = int(-(-n**0.5//1))+1 if m is not None: M = min(m+1, M) for i in range(2, M+1): if tmp<i: break if tmp%i==0: cnt=0 while tmp%i==0: cnt+=1 tmp //= i f[i] = cnt if tmp!=1: f[tmp] = 1 if not f: f[n] = 1 return f def primitive_root(p): if p == 2: return 1 if p == 167772161: return 3 if p == 469762049: return 3 if p == 754974721: return 11 if p == 998244353: return 3 g = 2 f = factor(p-1) while True: if all(pow(g,(p-1)//k,p)!=1 for k in f.keys()): break g += 1 return g # FFT import numpy as np TYPE = np.int64 M = 998244353 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) % 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 p = int(input()) a = list(map(int, input().split())) b = list(map(int, input().split())) g = primitive_root(p) n = p-1 ap = [0]*n bp = [0]*n v = 1 for i in range(n): ap[i] += a[v-1] bp[i] += b[v-1] v *= g v %= p cp = fft_large(ap,bp).tolist() c = [0]*n v = 1 for i in range(n): c[v-1] += cp[i] if len(cp)>i+n: c[v-1] += cp[i+n] c[v-1] %= M v *= g v %= p write(" ".join(map(str, c)))