use bump_allocator in many_facts

adamant-pwn · adamant-pwn · commit c8d763d94c6a · 2025-05-26T20:51:20.000+02:00
diff --git a/cp-algo/structures/eertree.hpp b/cp-algo/structures/eertree.hpp
@@ -5,7 +5,6 @@
 #include <iostream>
 #include <vector>
 #include <string>
-#include "../util/bump_alloc.hpp"
 namespace cp_algo::structures {
     template<int sigma = 26, char mch = 'a'>
     struct eertree {
@@ -63,7 +62,7 @@ namespace cp_algo::structures {
             print(std::identity{});
         }
     private:
-        std::vector<std::forward_list<int, bump_alloc<int>>> to;
+        std::vector<std::forward_list<int>> to;
         std::vector<int> len, link, par;
         std::string s;
         int n = 1, sz = 2, last = 0;
diff --git a/cp-algo/util/bump_alloc.hpp b/cp-algo/util/bump_alloc.hpp
@@ -1,10 +1,12 @@
 #ifndef CP_ALGO_UTIL_BUMP_ALLOC_HPP
 #define CP_ALGO_UTIL_BUMP_ALLOC_HPP
 #include <cstddef>
+#include "big_alloc.hpp"
 namespace cp_algo {
-    alignas(64) char buf[450 << 20];
-    size_t buf_ind = sizeof buf;
-    template<class T> struct bump_alloc {
+    template<class T, size_t max_len>
+    struct bump_alloc {
+        static char* buf;
+        static size_t buf_ind;
         using value_type = T;
         bump_alloc() = default;
         template<class U> bump_alloc(const U&) {}
@@ -15,5 +17,9 @@ namespace cp_algo {
         }
         void deallocate(T*, size_t) {}
     };
+    template<class T, size_t max_len>
+    char* bump_alloc<T, max_len>::buf = big_alloc<char>().allocate(max_len * sizeof(T));
+    template<class T, size_t max_len>
+    size_t bump_alloc<T, max_len>::buf_ind = max_len * sizeof(T);
 }
 #endif // CP_ALGO_UTIL_BUMP_ALLOC_HPP
diff --git a/verify/simd/many_facts.test.cpp b/verify/simd/many_facts.test.cpp
@@ -6,6 +6,7 @@
 #include "blazingio/blazingio.min.hpp"
 #include "cp-algo/util/checkpoint.hpp"
 #include "cp-algo/util/simd.hpp"
+#include "cp-algo/util/bump_alloc.hpp"
 #include "cp-algo/math/common.hpp"
 
 using namespace std;
@@ -14,13 +15,16 @@ using namespace cp_algo;
 constexpr int mod = 998244353;
 constexpr int imod = -math::inv2(mod);
 
+template<int maxn = 100'000>
 vector<int> facts(vector<int> const& args) {
     constexpr int accum = 4;
     constexpr int simd_size = 8;
     constexpr int block = 1 << 18;
     constexpr int subblock = block / simd_size;
-    static basic_string<array<int, 2>> odd_args_per_block[mod / subblock];
-    static basic_string<array<int, 2>> reg_args_per_block[mod / subblock];
+    using T = array<int, 2>;
+    using alloc = bump_alloc<T, 30 * maxn>;
+    basic_string<T, char_traits<T>, alloc> odd_args_per_block[mod / subblock];
+    basic_string<T, char_traits<T>, alloc> reg_args_per_block[mod / subblock];
     constexpr int limit_reg = mod / 64;
     int limit_odd = 0;
 
