support base hashing in primality testing, typedef big_vector

Author: adamant-pwn

cp-algo/number_theory/primality.hpp

@@ -26,11 +26,17 @@ namespace cp_algo::math {
             }
             return x == -1;
         };
-        return base::with_mod(m, [&](){
-            // Works for all m < 2^64: https://miller-rabin.appspot.com
-            return std::ranges::all_of(std::array{
-                2, 325, 9375, 28178, 450775, 9780504, 1795265022
-            }, test);
+        return base::with_mod(m, [&]() {
+#ifdef CP_ALGO_NUMBER_THEORY_PRIMALITY_BASES_HPP
+            uint16_t base2 = 7, base3 = 61;
+            if (m != uint32_t(m)) {
+                base2 = base_table1[uint32_t(m * 0xAD625B89) >> 18];
+                base3 = base_table2[base2 >> 13];
+            }
+            return test(2) && test(base2) && test(base3);
+#else
+            return std::ranges::all_of(std::array{2, 325, 9375, 28178, 450775, 9780504, 1795265022}, test);
+#endif
         });
     }
 }

cp-algo/number_theory/primality_bases.hpp

@@ -1,6 +1,7 @@
 #ifndef CP_ALGO_UTIL_big_alloc_HPP
 #define CP_ALGO_UTIL_big_alloc_HPP
 
+#include <vector>
 #include <cstddef>
 #include <iostream>
 
@@ -59,5 +60,8 @@ namespace cp_algo {
             return (x + Align - 1) / Align * Align;
         }
     };
+
+    template<typename T>
+    using big_vector = std::vector<T, big_alloc<T>>;
 }
 #endif // CP_ALGO_UTIL_big_alloc_HPP

cp-algo/util/big_alloc.hpp

@@ -1,6 +1,9 @@
 // @brief Primality Test
 #define PROBLEM "https://judge.yosupo.jp/problem/primality_test"
 #pragma GCC optimize("Ofast,unroll-loops")
+#include <iostream>
+#include "blazingio/blazingio.min.hpp"
+#include "cp-algo/number_theory/primality_bases.hpp"
 #include "cp-algo/number_theory/primality.hpp"
 #include <bits/stdc++.h>