My_CP_Library/Lichao_Tree(Segment).cpp at main · ngducmanh21/My_CP_Library · GitHub

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
struct Segment_Tree {
	int n;
	struct Line {
		long long a; long long b;

		long long get(int x) {
			return a * x + b;
		}
	} seg;

	vector <Line> it;

	Segment_Tree(int n) {
		this->n = n;
		it.resize(4 * n + 7);
	}

	void update(int i, int l, int r, int u, int v, Line &seg) {
		if (l > v || r < u) return;
		int mid = (l + r) >> 1;

		if (u <= l && r <= v) {
			// Y MAX

			if (it[i].get(l) >= seg.get(l) && it[i].get(r) >= seg.get(r)) {
				return;
			}

			if (it[i].get(l) <= seg.get(l) && it[i].get(r) <= seg.get(r)) {
				it[i] = seg;
				return;
			}

			if (it[i].get(l) >= seg.get(l) && it[i].get(mid) >= seg.get(mid)) {
				update(i << 1 | 1, mid + 1, r, u, v, seg);
				return;
			}

			if (it[i].get(l) <= seg.get(l) && it[i].get(mid) <= seg.get(mid)) {
				update(i << 1 | 1, mid + 1, r, u, v, it[i]);
				it[i] = seg;
				return;
			}

			if (it[i].get(r) >= seg.get(r) && it[i].get(mid + 1) >= seg.get(mid + 1)) {
				update(i << 1, l, mid, u, v, seg);
				return;
			}

			if (it[i].get(r) <= seg.get(r) && it[i].get(mid + 1) <= seg.get(mid + 1)) {
				update(i << 1, l, mid, u, v, it[i]);
				it[i] = seg;
				return;
			}

			return;
		}

		update(i << 1, l, mid, u, v, seg);
		update(i << 1 | 1, mid + 1, r, u, v, seg);
	}

	long long get(int i, int l, int r, int u) {
		if (l > u || r < u) return 0;
		long long res = it[i].get(u);
		if (l == r) return res;
		int mid = (l + r) >> 1;
		maximize(res, get(i << 1, l, mid, u));
		maximize(res, get(i << 1 | 1, mid + 1, r, u));
		return res;
	}
};