Update correct.cpp

yosupo06 · Jan 27, 2025 · d782eca · d782eca
1 parent e5610a0
commit d782eca
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tree/vertex_add_path_sum/sol/correct.cpp b/tree/vertex_add_path_sum/sol/correct.cpp
@@ -72,7 +72,8 @@ template <class E> struct HLExec : HL {
     V<int> tch;
     int id = 0;
     HLExec(const VV<E>& _g, int r) : HL(_g.size()), g(_g), tch(g.size(), -1) {
-        assert(dfs_sz(r, -1) == int(g.size()));
+        int size = dfs_sz(r, -1);
+        assert(size == int(g.size()));
         dfs(r, -1);
         init_extra();
     }

diff --git a/tree/vertex_set_path_composite/sol/correct.cpp b/tree/vertex_set_path_composite/sol/correct.cpp
@@ -79,7 +79,8 @@ template <class E> struct HLExec : HL {
     V<int> tch;
     int id = 0;
     HLExec(const VV<E>& _g, int r) : HL(_g.size()), g(_g), tch(g.size(), -1) {
-        assert(dfs_sz(r, -1) == int(g.size()));
+        int size = dfs_sz(r, -1);
+        assert(size == int(g.size()));
         dfs(r, -1);
         init_extra();
     }