forked from Ashishgup1/Competitive-Coding
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Segment Tree.cpp
150 lines (134 loc) · 2.79 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
147
148
149
150
struct data
{
//Use required attributes
int mn;
//Default Values
data() : mn(1e9) {};
};
struct SegTree
{
int N;
vector<data> st;
vector<bool> cLazy;
vector<int> lazy;
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
void merge(data &cur, data &l, data &r)
{
cur.mn = min(l.mn, r.mn);
}
//Handle lazy propagation appriopriately
void propagate(int node, int L, int R)
{
if(L != R)
{
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];
cLazy[node] = 0;
}
void build(int node, int L, int R)
{
if(L==R)
{
st[node].mn = 1e9;
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, 1, N, pos);
}
data query(int l, int r)
{
return Query(1, 1, N, l, r);
}
void update(int pos, int val)
{
pUpdate(1, 1, N, pos, val);
}
void update(int l, int r, int val)
{
Update(1, 1, N, l, r, val);
}
};
//Problem 1 (Max Query - Point Update with Coordinate Compression): http://codeforces.com/gym/100733/problem/F
//Solution 1: http://codeforces.com/gym/100733/submission/41643795
//Problem 2 (Min Query - Offline processing): https://codeforces.com/problemset/problem/522/D
//Solution 2: https://codeforces.com/contest/522/submission/45493164