Shorter LineContainer

(thanks, dacin21!)
kth-competitive-programming · Dec 19, 2018 · 165807e · 165807e
1 parent 7152840
commit 165807e
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/content/data-structures/LineContainer.h b/content/data-structures/LineContainer.h
@@ -10,15 +10,13 @@
  */
 #pragma once
 
-bool Q;
 struct Line {
 	mutable ll k, m, p;
-	bool operator<(const Line& o) const {
-		return Q ? p < o.p : k < o.k;
-	}
+	bool operator<(const Line& o) const { return k < o.k; }
+	bool operator<(ll x) const { return p < x; }
 };
 
-struct LineContainer : multiset<Line> {
+struct LineContainer : multiset<Line, less<>> {
 	// (for doubles, use inf = 1/.0, div(a,b) = a/b)
 	const ll inf = LLONG_MAX;
 	ll div(ll a, ll b) { // floored division
@@ -38,7 +36,7 @@ struct LineContainer : multiset<Line> {
 	}
 	ll query(ll x) {
 		assert(!empty());
-		Q = 1; auto l = *lower_bound({0,0,x}); Q = 0;
+		auto l = *lower_bound(x);
 		return l.k * x + l.m;
 	}
 };