Skip to content

Commit 1e64160

Browse files
authored
Update shallowest_decomp_tree.hpp
1 parent 021ba44 commit 1e64160

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

library/trees/shallowest_decomp_tree.hpp

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,11 +21,12 @@ void shallowest(auto& adj, auto f) {
2121
return dp;
2222
};
2323
dfs(dfs, 0, 0);
24-
for (vi vec : order | views::reverse)
24+
auto cpy(adj);
25+
adj.assign(n, {});
26+
for (vi vec : order)
2527
for (int v : vec) {
2628
f(v);
27-
for (int u : adj[v])
28-
iter_swap(ranges::find(adj[u], v), rbegin(adj[u])),
29-
adj[u].pop_back();
29+
adj[v] = cpy[v];
30+
for (int u : adj[v]) adj[u].push_back(v);
3031
}
3132
}

0 commit comments

Comments
 (0)