[백준 2132번] 나무 위의 벌레

문제 링크: https://www.acmicpc.net/problem/2132

2132번: 나무 위의 벌레

<문제 풀이> DFS, BFS, 트리의 지름

 

각 정점에서 BFS를 돌려서 각 정점에서 시작했을 때 최대 거리를 구합니다.

최대 거리가 가장 큰 정점을 시작 정점으로 하면 됩니다.

+트리의 지름을 구하는 것과 동일합니다.

 

 

<C++ 소스 코드>

#include <iostream>
#include <string>
#include <algorithm>
#include <queue>
#include <vector>
#include <stack>
#include <utility>
#include <climits>
#include <deque>
using namespace std;
 
int n;
int vertex[10001];
vector<int> adj[10001];
int dist[10001];
int max_cnt = 0;
int start = 0;
void bfs() {
    for (int i = n; i >=1; i--) {
        fill(dist, dist + 10001, -1);
        queue<int> Q;
        Q.push(i);
        dist[i] = vertex[i];
        while (!Q.empty()) {
            int cur = Q.front(); Q.pop();
            if (dist[cur] >= max_cnt) {
                max_cnt = dist[cur];
                start = i;
            }
            for (auto& next : adj[cur]) {
                if (dist[next] != -1) continue;
                dist[next] = dist[cur] + vertex[next];
                Q.push(next);
            }
        }
    }
 
}
 
int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
 
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> vertex[i];
    }
    for (int i = 0; i < n - 1; i++) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    bfs();
    cout << max_cnt << " " << start;
    return 0;
}