가장 먼 노드

문제 설명

n개의 노드가 있는 그래프가 있다.

각 노드는 1부터 n까지 번호가 적혀있다.

1번 노드에서 가장 멀리 떨어진 노드의 갯수를 구하려고 한다.

가장 멀리 떨어진 노드란 최단경로로 이동했을 때 간선의 개수가 가장 많은 노드들을 의미한다.

 

노드의 개수 n, 간선에 대한 정보가 담긴 2차원 배열 vertex가 매개변수로 주어질 때, 1번 노드로부터 가장 멀리 떨어진 노드가 몇 개인지를 return 하도록 solution 함수를 작성하라.

 

제한사항

노드의 개수 n은 2 이상 20,000 이하입니다.

간선은 양방향이며 총 1개 이상 50,000개 이하의 간선이 있습니다.

(vertex 배열 각 행 [a, b]는 a번 노드와 b번 노드 사이에 간선이 있다는 의미)

입출력 예

n	vertex	return
6	[[3, 6], [4, 3], [3, 2], [1, 3], [1, 2], [2, 4], [5, 2]]	3

 

---

첫 시도

#include <string>
#include <vector>
#include <algorithm>
#include <queue>

using namespace std;

int solution(int n, vector<vector<int>> edge) {
    int answer = 0;
    vector<vector<int>> graph(n+1);
    vector<int> distance(n + 1, -1); 
    
    for(vector<int>& def : edge)
    {
        graph[def[0]].push_back(def[1]);
        graph[def[1]].push_back(def[0]);
    }
    
    queue<int> q;
    q.push(1); // 시작 노드는 1번 노드
    distance[1] = 0;

    while (!q.empty()) {
        int current = q.front();
        q.pop();

        for (int neighbor : graph[current]) {
            if (distance[neighbor] == -1) {
                distance[neighbor] = distance[current] + 1;
                q.push(neighbor);
            }
        }
    }

    int maxDistance = *max_element(distance.begin(), distance.end());

    // 가장 먼 거리에 있는 노드들의 개수 세기
    return count(distance.begin(), distance.end(), maxDistance);
}

성공

그래프를 생성하고 edge를 재정렬해준다.

 

1번에서 1번 2번에 2번이 들어가도록 정리.

 

큐를 한개 생성하고, 해당 큐에 1번이 시작점 임으로 시작노드를 정의.

distance는 1번의 거리가 0이라고 함.

 

반복문을 통해 그래프의 데이터를 전부 순환하면서, 해당 그래프의 거리를 계속 순환한다.

거리는 생성때 -1로 생성되었음 으로 방문하지 않았다고 -1이라고 한다.

 

방문하지 않은 노드 였을 때 디스턴스의 거리 = 현재 지정된 탐색우선순위인 current의 거리에 1을 더해서 정의해준다.

1번이 1번을 볼 경우, 거리가 0, 즉, 1번은 1번과 거리가 0이라는의미.

 

그리고, 연결된 neighbor를 큐에 넣어서 계속 탐색한다.

 

탐색을 못할 때 까지 탐색.

 

그리고, 가장 큰 값을 distance에서 찾는다.

 

그리고, 가장 큰 값을 거리에서 전부 새어서 리턴한다.