[Baekjoon] 1967 트리의 지름 (C++ 풀이)

https://www.acmicpc.net/problem/1967

#include <cstring>
#include <iostream>
#include <utility>
#include <vector>

using namespace std;
using pii = pair<int, int>;
using vpii = vector<pii>;

#define MAX 10'001

int N;
vpii MAP[MAX];
bool visited[MAX];
int max_dist = 0;
int fst_max_node;

void dfs(int start, int dist) {
    if (visited[start])
        return;
    if (max_dist < dist) {
        max_dist = dist;
        fst_max_node = start;
    }
    visited[start] = true;
    for (int j = 0; j < MAP[start].size(); j++) {
        int next = MAP[start][j].first;
        int next_dist = MAP[start][j].second;
        dfs(next, next_dist + dist);
    }
}

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

    cin >> N;
    N--;

    int u, v, l;
    while (N--) {
        cin >> u >> v >> l;
        MAP[u].push_back({v, l});
        MAP[v].push_back({u, l});
    }

    dfs(1, 0);
    memset(visited, 0, sizeof(visited));
    max_dist = 0;
    dfs(fst_max_node, 0);
    cout << max_dist;

    return 0;
}