fix test

hitonanode · hitonanode · commit ca9154c5ab49 · 2024-09-25T00:45:51.000+09:00
diff --git a/tree/centroid_decomposition.hpp b/tree/centroid_decomposition.hpp
@@ -14,7 +14,7 @@ struct CentroidDecomposition {
     std::vector<int> is_alive;
     std::vector<int> subtree_size;
 
-    void decompose(int r, int conn_size, auto callback) {
+    template <class F> void decompose(int r, int conn_size, F callback) {
 
         const int c = find_current_centroids(r, conn_size).first;
         is_alive.at(c) = 0;
@@ -69,7 +69,7 @@ struct CentroidDecomposition {
         return {c1, c2};
     }
 
-    void run(int r, auto callback) {
+    template <class F> void run(int r, F callback) {
         int conn_size = 0;
 
         auto rec = [&](auto &&self, int now, int prv) -> void {
diff --git a/tree/frequency_table_of_tree_distance.hpp b/tree/frequency_table_of_tree_distance.hpp
@@ -22,7 +22,15 @@ struct frequency_table_of_tree_distance {
     }
     frequency_table_of_tree_distance(const std::vector<std::vector<int>> &to) {
         tos = to;
-        cd = CentroidDecomposition(to).centroid_decomposition(0);
+
+        CentroidDecomposition c(to.size());
+        for (int i = 0; i < int(to.size()); i++) {
+            for (int j : to[i]) {
+                if (i < j) c.add_edge(i, j);
+            }
+        }
+
+        cd = c.centroid_decomposition(0);
     }
     template <class S, std::vector<S> (*conv)(const std::vector<S> &, const std::vector<S> &)>
     std::vector<S> solve(const std::vector<S> &weight) {
