究极抽象压行,1行FFT/NTT
  • 板块灌水区
  • 楼主bcdmwSjy
  • 当前回复0
  • 已保存回复0
  • 发布时间2023/10/12 10:15
  • 上次更新2023/11/2 14:21:01
查看原帖
究极抽象压行,1行FFT/NTT
514727
bcdmwSjy楼主2023/10/12 10:15

P3803 【模板】多项式乘法(FFT)

用到了python的第三方模块sympy

import sympy
#FFT
print(input()*0+' '.join(map(str,sympy.ifft(((c:=sympy.fft((a:=list(map(int,input().split())))+[0]*len(b:=list(map(int,input().split()))))),(d:=sympy.fft(b+[0]*len(a))),[i*j for i,j in zip(c,d)])[2])[:len(a)+len(b)-1])))

#NTT
print(input()*0+' '.join(map(str,sympy.intt(((c:=sympy.ntt((a:=list(map(int,input().split())))+[0]*len(b:=list(map(int,input().split()))),998244353)),(d:=sympy.ntt(b+[0]*len(a),998244353)),[i*j%998244353 for i,j in zip(c,d)])[2],998244353)[:len(a)+len(b)-1])))

不过洛谷不让用sympy啊...

2023/10/12 10:15
加载中...