`assert 1 <= n` になってますが,本家では $0\leq N$ になっていて,そちらに合わせるのが自然だと思います.(おそらくタイポ?) https://atcoder.github.io/ac-library/master/document_ja/math.html#:~:text=32-,0%E2%89%A4,32,-1