fix bugs for lazy_segment_tree and it's test

Mr-Python-in-China · Oct 12, 2024 · 807a3c7 · 807a3c7
1 parent 5b61ed8
commit 807a3c7
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/.github/workflows/testLibrary.yml b/.github/workflows/testLibrary.yml
@@ -30,4 +30,4 @@ jobs:
       - name: Test
         run: |
           cd library
-          ./test.exe --gtest_repeat=8
+          ./test.exe --gtest_repeat=20
diff --git a/library/makefile b/library/makefile
@@ -2,7 +2,7 @@ STDCXX ?= c++14
 
 CXX ?= g++
 
-CXXFLAGS = -Wall -Wextra -g -I. -MMD -std=$(STDCXX)
+CXXFLAGS = -Wall -Wextra -g -I. -MMD -std=$(STDCXX) -O2
 
 TARGET = test.exe
 

diff --git a/library/mrpython/lazy_segment_tree.hpp b/library/mrpython/lazy_segment_tree.hpp
@@ -103,6 +103,7 @@ class lazy_segment_tree {
         n(len),
         lazyInit(lazyInitVal),
         mergeData(mergeDataFun),
+        operate(OperateFun),
         mergeLazy(mergeTagFun) {
     build();
   }

diff --git a/library/test/lazy_segment_tree/lazy_segment_tree_add_add.cpp b/library/test/lazy_segment_tree/lazy_segment_tree_add_add.cpp
@@ -19,12 +19,13 @@ TEST(lazy_segment_tree, add) {
     if (l > r) std::swap(l, r);
     assert(l < r + 1);
     if (operator_dist(gen)) {
-      int ans = std::accumulate(a.begin() + l, a.begin() + r + 1, (unsigned)0);
+      unsigned ans =
+          std::accumulate(a.begin() + l, a.begin() + r + 1, (unsigned)0);
       EXPECT_EQ(tree.get(l, r + 1), ans);
     } else {
       unsigned value = val_dist(gen);
-      for (std::size_t i = l; i < r; ++i) a[i] += value;
-      tree.set(l, r, value);
+      for (std::size_t i = l; i < r + 1; ++i) a[i] += value;
+      tree.set(l, r + 1, value);
     }
   }
 }