Update Segment Tree.cpp

Segment Tree.cpp

@@ -1,46 +1,146 @@
-Min Query (Point Update):
+struct data
+{
+	//Use required attributes
+	int mn;
 
-int st[4*N];
+	//Default Values
+	data() : mn(1e9) {};
+};
 
-void build(int node, int L, int R)
+struct SegTree
 {
-	if(L==R)
+	int N;
+	vector<data> st;
+	vector<bool> cLazy;
+	vector<int> lazy;
+
+	void init(int n)
 	{
-		st[node]=1e9;
-		return;
+		N = n;
+		st.resize(4 * N + 5);
+		cLazy.assign(4 * N + 5, false);
+		lazy.assign(4 * N + 5, 0);
 	}
-	int M=(L+R)/2;
-	build(node*2, L, M);
-	build(node*2+1, M+1, R);
-	st[node]=min(st[node*2], st[node*2+1]);
-}
 
-int query(int node, int L, int R, int i, int j)
-{
-	if(j<L || i>R)
-		return 1e9;
-	if(i<=L && R<=j)
-		return st[node];
-	int M=(L+R)/2;
-	int left=query(node*2, L, M, i, j);
-	int right=query(node*2 + 1, M+1, R, i, j);
-	return min(left, right);
-}
-
-void update(int node, int L, int R, int pos, int val)
-{
-	if(L==R)
-	{
-		st[node]=val;
-		return;
-	}
-	int M=(L+R)/2;
-	if(pos<=M)
-		update(node*2, L, M, pos, val);
-	else
-		update(node*2 + 1, M+1, R, pos, val);
-	st[node]=min(st[node*2], st[node*2 + 1]);
-}
+	//Write reqd merge functions
+	void merge(data &cur, data &l, data &r) 
+	{
+		cur.mn = min(l.mn, r.mn);
+	}
+
+	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, 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