Skip to content

Commit 02ae4c6

Browse files
committed
another golf
1 parent b8bd506 commit 02ae4c6

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

library/trees/centroid_decomp.hpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,18 @@
66
//! @time O(n log n)
77
//! @space O(n)
88
vi cd(auto& g, auto f) {
9-
vi p(sz(g), -1), siz(sz(g), sz(g));
9+
vi p(sz(g), -1), s(sz(g), sz(g));
1010
auto ctd = [&](auto ctd, int u, int p, int n) -> int {
11-
siz[u] = 1;
11+
s[u] = 1;
1212
for (int v : g[u])
1313
if (v != p) {
1414
if (int c = ctd(ctd, v, u, n); c != -1) return c;
15-
siz[u] += siz[v];
15+
s[u] += s[v];
1616
}
17-
return 2 * siz[u] >= n ? siz[p] = n - siz[u], u : -1;
17+
return 2 * s[u] >= n ? s[p] = n - s[u], u : -1;
1818
};
1919
auto dfs = [&](auto dfs, int u) -> int {
20-
f(u = ctd(ctd, u, u, siz[u]));
20+
f(u = ctd(ctd, u, u, s[u]));
2121
for (int v : g[u]) erase(g[v], u), p[dfs(dfs, v)] = u;
2222
return u;
2323
};

0 commit comments

Comments
 (0)