From 4979378b4b22d1db9a972e7f576cdcb94d79e7e0 Mon Sep 17 00:00:00 2001
From: Gloria Mundi <gloria@gloria-mundi.eu>
Date: Tue, 27 Feb 2024 23:04:21 +0100
Subject: golf binary lifting

---
 graph/binary_lifting.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'graph/binary_lifting.cpp')

diff --git a/graph/binary_lifting.cpp b/graph/binary_lifting.cpp
index 9e0dfc3..0b8c218 100644
--- a/graph/binary_lifting.cpp
+++ b/graph/binary_lifting.cpp
@@ -20,8 +20,8 @@ struct Lift {
 	int lca(int u, int v) {
 		v = lift(v, dep[u]), u = lift(u, dep[v]);
 		while (u != v) {
-			if (jmp[u] == jmp[v]) u = par[u], v = par[v];
-			else u = jmp[u], v = jmp[v];
+			auto &a = jmp[u] == jmp[v] ? par : jmp;
+			u = a[u], v = a[v];
 		}
 		return u;
 	}
-- 
cgit v1.2.3