From a24ce98f2c79d5594d171a9b294b1cf25b488ebc Mon Sep 17 00:00:00 2001
From: Gloria Mundi <gloria@gloria-mundi.eu>
Date: Wed, 1 May 2024 18:07:15 +0200
Subject: improve test system, fix segment tree, add more segment tree tests

---
 datastructures/test/segmentTree.awk | 74 +++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 datastructures/test/segmentTree.awk

(limited to 'datastructures/test/segmentTree.awk')

diff --git a/datastructures/test/segmentTree.awk b/datastructures/test/segmentTree.awk
new file mode 100644
index 0000000..e863d4e
--- /dev/null
+++ b/datastructures/test/segmentTree.awk
@@ -0,0 +1,74 @@
+
+/Neutral element for combine/ {
+	print "#ifndef SEGTREE_MUL"
+	print "# ifndef SEGTREE_FIRST_NEG"
+	print
+	print "# else"
+	tmp = $0
+	sub(/0/, "numeric_limits<ll>::max()", tmp)
+	print tmp
+	print "# endif"
+	print "#else"
+	sub(/0/, "1")
+	print
+	print "#endif"
+	next
+}
+
+/modify this \+ neutral/ {
+	print "#ifndef SEGTREE_MUL"
+	print "# ifndef SEGTREE_FIRST_NEG"
+	print
+	print "# else"
+	tmp = $0
+	sub(/a \+ b/, "a < 0 ? a : min(a, b)", tmp)
+	print tmp
+	print "# endif"
+	print "#else"
+	sub(/a \+ b/, "a*b % MOD")
+	print
+	print "#endif"
+	next
+}
+
+/SegTree\(vector<T>& a\)/ {
+	print "#ifndef SEGTREE_INIT_DEFAULT"
+	print
+	getline
+	print
+	print "#else"
+	getline
+	sub(/\/\//, "")
+	print
+	getline
+	sub(/\/\//, "")
+	print
+	print "#endif"
+	next
+}
+
+/remove for range update/ {
+	print "#ifndef SEGTREE_RANGE_UPDATE"
+	print
+	getline
+	print
+	getline
+	print "\t\t}"
+	print "#endif"
+	print "\t}"
+	next
+}
+
+/void update/ {
+	print "#ifndef SEGTREE_RANGE_UPDATE"
+}
+
+/OR: range update/ {
+	print "#else"
+}
+
+/^\};$/ {
+	print "#endif"
+}
+
+{ print }
-- 
cgit v1.2.3