338. Familystrokes May 2026

root = 1 stack = [(root, 0)] # (node, parent) internal = 0 horizontal = 0

Proof. If childCnt ≥ 2 : the children occupy at least two columns on the next row, so a horizontal line is needed to connect the leftmost to the rightmost child (rule 2).

const int ROOT = 1; vector<int> parent(N + 1, 0); vector<int> st; // explicit stack for DFS st.reserve(N); st.push_back(ROOT); parent[ROOT] = -1; // mark visited 338. FamilyStrokes

while stack not empty: v, p = pop(stack) childCnt = 0 for each w in G[v]: if w == p: continue // ignore the edge back to parent childCnt += 1 push (w, v) on stack

1 if childCnt(v) = 1 2 if childCnt(v) ≥ 2 0 if childCnt(v) = 0 Proof. Directly from Lemma 2 (vertical) and Lemma 3 (horizontal). ∎ answer = internalCnt + horizontalCnt computed by the algorithm equals the minimum number of strokes needed to draw the whole tree. root = 1 stack = [(root, 0)] #

if childCnt > 0: // v has at least one child → internal internalCnt += 1 if childCnt >= 2: horizontalCnt += 1

int main() ios::sync_with_stdio(false); cin.tie(nullptr); int N; if (!(cin >> N)) return 0; vector<vector<int>> g(N + 1); for (int i = 0, u, v; i < N - 1; ++i) cin >> u >> v; g[u].push_back(v); g[v].push_back(u); Directly from Lemma 2 (vertical) and Lemma 3 (horizontal)

Proof. The drawing rules require a vertical line from the node down to the row of its children whenever it has at least one child. The line is mandatory and unique, hence exactly one vertical stroke. ∎ An internal node requires a horizontal stroke iff childCnt ≥ 2 .