Update bts-hotcold.mdx

cpinitiative · May 5, 2024 · deaad3a · deaad3a
1 parent 0710999
commit deaad3a
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/solutions/platinum/bts-hotcold.mdx b/solutions/platinum/bts-hotcold.mdx
@@ -81,6 +81,17 @@ class Tree {
 		tout[u] = timer - 1;
 	}
 
+	void update(int u, int v, int initial, int change) {
+		diff[u][0] += initial;
+		diff[u][1] += change;
+		if (change > 0) {
+			diff[v][0] -= initial + (depth[u] - depth[v]);
+		} else {
+			diff[v][0] -= initial - (depth[u] - depth[v]);
+		}
+		diff[v][1] -= change;
+	}
+
 	void dfs(int u, int p) {
 		for (int v : adj[u]) {
 			if (v == p) { continue; }
@@ -121,17 +132,6 @@ class Tree {
 		return depth[u] + depth[v] - 2 * depth[anc];
 	}
 
-	void update(int u, int v, int initial, int change) {
-		diff[u][0] += initial;
-		diff[u][1] += change;
-		if (change > 0) {
-			diff[v][0] -= initial + (depth[u] - depth[v]);
-		} else {
-			diff[v][0] -= initial - (depth[u] - depth[v]);
-		}
-		diff[v][1] -= change;
-	}
-
 	void query(int a, int b, int t) {
 		int anc = lca(a, b);
 		if (anc == t || !is_ancestor(anc, t)) {