Fix

adamant-pwn · adamant-pwn · commit ee5a871f0666 · 2024-10-07T21:21:13.000+02:00
diff --git a/cp-algo/number_theory/modint.hpp b/cp-algo/number_theory/modint.hpp
@@ -41,7 +41,7 @@ namespace cp_algo::math {
         }
         modint_base(): r(0) {}
         modint_base(int64_t rr): r(rr % mod()) {
-            r = std::min(r, r + 2 * mod());
+            r = std::min(r, r + mod());
             r = m_transform(r);
         }
         modint inv() const {
@@ -99,7 +99,7 @@ namespace cp_algo::math {
             return std::min(res, res - mod());
         }
         void setr_direct(uint64_t rr) {r = rr;}
-        uint64_t getr_direct() const {return r;}
+        uint64_t getr_direct() const {return std::min(r, r - mod());}
     private:
         uint64_t r;
         modint& to_modint() {return static_cast<modint&>(*this);}

Original file line number	Diff line number	Diff line change
`@@ -41,7 +41,7 @@ namespace cp_algo::math {`
`41`	`41`	`}`
`42`	`42`	`modint_base(): r(0) {}`
`43`	`43`	`modint_base(int64_t rr): r(rr % mod()) {`
`44`		`- r = std::min(r, r + 2 * mod());`
	`44`	`+ r = std::min(r, r + mod());`
`45`	`45`	`r = m_transform(r);`
`46`	`46`	`}`
`47`	`47`	`modint inv() const {`
`@@ -99,7 +99,7 @@ namespace cp_algo::math {`
`99`	`99`	`return std::min(res, res - mod());`
`100`	`100`	`}`
`101`	`101`	`void setr_direct(uint64_t rr) {r = rr;}`
`102`		`- uint64_t getr_direct() const {return r;}`
	`102`	`+ uint64_t getr_direct() const {return std::min(r, r - mod());}`
`103`	`103`	`private:`
`104`	`104`	`uint64_t r;`
`105`	`105`	`modint& to_modint() {return static_cast<modint&>(*this);}`