[1167] 트리의 지름 (C++)

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;
int max_cost = 0; int farthest = 1;
vector<pair<int, int>> A[100002];

void DFS(int start, int prev, int cost);

int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int v; cin >> v;
    int a, b, c;

    for (int i = 0; i < v; i++) {
        cin >> a;
        while (1) {
            cin >> b;
            if (b == -1) break;
            cin >> c;
            A[a].push_back(make_pair(b, c));
        }
    }

    DFS(1, 0, 0);
    max_cost = 0;
    DFS(farthest, 0, 0);
    cout << max_cost;

    return 0;
}

void DFS(int start, int prev, int cost) {
    if (cost > max_cost) {
        max_cost = cost;
        farthest = start;
    }

    for (int i = 0; i < A[start].size(); i++) {
        if (A[start][i].first != prev)
            DFS(A[start][i].first, start, cost + A[start][i].second);
    }
}

 

1. 루트 노드에서 가장 먼 두 노드 사이의 거리가 트리의 지름

2. 첫 번째 DFS에서 기본 루트를 1로 설정하고, 가장 먼 노드를 찾음

3. 두 번째 DFS에서는 2에서 찾은 노드를 기준으로 가장 먼 노드를 다시 탐색

4. max_cost에 두 노드 사이의 거리를 저장 후 출력