summaryrefslogtreecommitdiff
path: root/datastructures/test/segmentTree.awk
blob: e863d4e378b3e9d1687b37476fe068d3fe0027b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
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 }