Skip to content

Commit a77a224

Browse files
committed
golf
1 parent 3b21c06 commit a77a224

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

library/trees/centroid_decomp.hpp

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -9,21 +9,21 @@
99
template<class F> struct centroid {
1010
vector<vi> adj;
1111
F f;
12-
vi sub_sz;
12+
vi siz;
1313
centroid(const vector<vi>& a_adj, F a_f):
14-
adj(a_adj), f(a_f), sub_sz(sz(adj), -1) {
15-
rep(i, 0, sz(adj)) if (sub_sz[i] == -1) dfs(i, -1);
14+
adj(a_adj), f(a_f), siz(sz(adj), -1) {
15+
rep(i, 0, sz(adj)) if (siz[i] == -1) dfs(i, -1);
1616
}
1717
void calc_sz(int v, int p) {
18-
sub_sz[v] = 1;
18+
siz[v] = 1;
1919
for (int u : adj[v])
20-
if (u != p) calc_sz(u, v), sub_sz[v] += sub_sz[u];
20+
if (u != p) calc_sz(u, v), siz[v] += siz[u];
2121
}
2222
void dfs(int v, int p) {
2323
calc_sz(v, -1);
24-
for (int w = -1, sz_root = sub_sz[v];;) {
24+
for (int w = -1, sz_root = siz[v];;) {
2525
auto big_ch = find_if(all(adj[v]), [&](int u) {
26-
return u != w && 2 * sub_sz[u] > sz_root;
26+
return u != w && 2 * siz[u] > sz_root;
2727
});
2828
if (big_ch == end(adj[v])) break;
2929
w = v, v = *big_ch;

0 commit comments

Comments
 (0)