cheran-senthil / PyRival

⚡ Competitive Programming Library
https://pyrival.readthedocs.io/
Apache License 2.0
1.15k stars 311 forks source link

NTT fails for n = 1 #76

Open dnialh opened 2 years ago

dnialh commented 2 years ago

If a and b both have size 1 then ntt will have a runtime error (or (0,2), (2,0) but I don't know if this case comes up). I'm not sure whether the best solution is to hardcode this case or to do something else.