2.18.1 Traversal & Subtree Size

#include <iostream>
#include <vector>

using namespace std;

vector<int> g[101];
int p[101], d[101], sz[101];

void dfs(int u, int par) {
    // 자식 탐색 전 부모와 크기 초기값 설정
    p[u] = par;
    sz[u] = 1;
    for (int v : g[u]) {
        if (v == par) continue;
        d[v] = d[u] + 1;
        dfs(v, u);
        // 자식 탐색 후 서브트리 크기 합산
        sz[u] += sz[v];
    }
}

int main() {
    int n, i, a, b;

    cin >> n;
    for (i = 0; i < n - 1; ++i) {
        cin >> a >> b;
        g[a].push_back(b);
        g[b].push_back(a);
    }

    dfs(1, 0);

    for (i = 1; i <= n; ++i) {
        cout << p[i] << ' ' << d[i] << ' ' << sz[i] << '\n';
    }
}

import java.util.*;

public class Main {
    static ArrayList<ArrayList<Integer>> g = new ArrayList<>();
    static int[] p = new int[101], d = new int[101], sz = new int[101];

    static void dfs(int u, int par) {
        // 자식 탐색 전 부모와 크기 초기값 설정
        p[u] = par;
        sz[u] = 1;
        for (int v : g.get(u)) {
            if (v == par) continue;
            d[v] = d[u] + 1;
            dfs(v, u);
            // 자식 탐색 후 서브트리 크기 합산
            sz[u] += sz[v];
        }
    }

    public static void main(String[] args) {
        Scanner sc;
        int n, i, a, b;

        sc = new Scanner(System.in);
        n = sc.nextInt();
        for (i = 0; i <= n; ++i) {
            g.add(new ArrayList<>());
        }
        for (i = 0; i < n - 1; ++i) {
            a = sc.nextInt();
            b = sc.nextInt();
            g.get(a).add(b);
            g.get(b).add(a);
        }

        dfs(1, 0);

        for (i = 1; i <= n; ++i) {
            System.out.println(p[i] + " " + d[i] + " " + sz[i]);
        }
    }
}

def dfs(u, par):
    # 자식 탐색 전 부모와 크기 초기값 설정
    p[u] = par
    sz[u] = 1
    for v in g[u]:
        if v == par:
            continue
        d[v] = d[u] + 1
        dfs(v, u)
        # 자식 탐색 후 서브트리 크기 합산
        sz[u] += sz[v]


n = int(input())
g = [[] for _ in range(n + 1)]
p = [0] * (n + 1)
d = [0] * (n + 1)
sz = [0] * (n + 1)
for _ in range(n - 1):
    a, b = map(int, input().split())
    g[a].append(b)
    g[b].append(a)

dfs(1, 0)

for i in range(1, n + 1):
    print(p[i], d[i], sz[i])