-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsegment_tree.cpp
146 lines (133 loc) · 2.86 KB
/
segment_tree.cpp
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
struct segtree
{
struct data
{
//Use required attributes
int mn;
//Default Values
data() : mn(1e9) {};
};
int N;
vector<data> st;
vector<bool> cLazy;//stores if this values needs to be pushed down or not.
vector<int> lazy;//stores the value that needs to be pushed down.
void init(int n)
{
N = n;
st.resize(4 * N + 5);
cLazy.assign(4 * N + 5, false);
lazy.assign(4 * N + 5, 0);
}
//Write reqd merge functions
//also consider what if data l or r has default values then what to return;
void merge(data &cur, data &l, data &r)
{
cur.mn = min(l.mn, r.mn);
}
//Handle lazy propagation appriopriately and carefully
//might need to do cLazy[node*2] = 1-cLazy[node] if updates cancels(like flip operation);
void propagate(int node, int L, int R)
{
if(L != R)//push the changes
{
cLazy[node*2] = 1;
cLazy[node*2 + 1] = 1;
lazy[node*2] = lazy[node];
lazy[node*2 + 1] = lazy[node];
}
st[node].mn = lazy[node];//then change the current node
cLazy[node] = 0;
lazy[node] = 0;
}
void build(int node, int L, int R)
{
if(L==R)
{
st[node].mn = 1e9;//change the value as per the question.
return;
}
int M=(L + R)/2;
build(node*2, L, M);
build(node*2 + 1, M + 1, R);
merge(st[node], st[node*2], st[node*2+1]);
}
data Query(int node, int L, int R, int i, int j)
{
if(cLazy[node])
propagate(node, L, R);
if(j<L || i>R)
return data();
if(i<=L && R<=j)
return st[node];
int M = (L + R)/2;
data left=Query(node*2, L, M, i, j);
data right=Query(node*2 + 1, M + 1, R, i, j);
data cur;
merge(cur, left, right);
return cur;
}
data pQuery(int node, int L, int R, int pos)
{
if(cLazy[node])
propagate(node, L, R);
if(L == R)
return st[node];
int M = (L + R)/2;
if(pos <= M)
return pQuery(node*2, L, M, pos);
else
return pQuery(node*2 + 1, M + 1, R, pos);
}
void Update(int node, int L, int R, int i, int j, int val)
{
if(cLazy[node])
propagate(node, L, R);
if(j<L || i>R)
return;
if(i<=L && R<=j)
{
cLazy[node] = 1;
lazy[node] = val;
propagate(node, L, R);
return;
}
int M = (L + R)/2;
Update(node*2, L, M, i, j, val);
Update(node*2 + 1, M + 1, R, i, j, val);
merge(st[node], st[node*2], st[node*2 + 1]);
}
void pUpdate(int node, int L, int R, int pos, int val)
{
if(cLazy[node])
propagate(node, L, R);
if(L == R)
{
cLazy[node] = 1;
lazy[node] = val;
propagate(node, L, R);
return;
}
int M = (L + R)/2;
if(pos <= M)
pUpdate(node*2, L, M, pos, val);
else
pUpdate(node*2 + 1, M + 1, R, pos, val);
merge(st[node], st[node*2], st[node*2 + 1]);
}
data query(int pos)
{
return pQuery(1, 0, N-1, pos);
}
data query(int l, int r)
{
return Query(1, 0, N-1, l, r);
}
void update(int pos, int val)
{
pUpdate(1, 0, N-1, pos, val);
}
void update(int l, int r, int val)
{
Update(1, 0, N-1, l, r, val);
}
};