Reroot codegolf

Author: enwask

content/graphs/Reroot.h

@@ -20,20 +20,15 @@ vector<T> dp, dpr;
 
 T dfs(int u, int p) {
 	dp[u] = init();
-	for (int v : adj[u]) {
-		if (v == p) continue;
+	for (int v : adj[u]) if (v != p)
 		dp[u] = combine(dp[u], dfs(v, u));
-	}
 	return dp[u];
 }
 
 T dfsr(int u, int p, T dpr_p = init()) {
 	dpr[u] = dp[u];
 	if (p != u) dpr[u] = combine(dpr[u], dpr_p);
-	for (int v : adj[u]) {
-		if (v == p) continue;
-		T dec = decombine(dpr[u], dp[v]);
-		dfsr(v, u, dec);
-	}
+	for (int v : adj[u]) if (v != p)
+		dfsr(v, u, decombine(dpr[u], dp[v]));
 	return dpr[u];
 }