문제
- n개의 노드가 있는 그래프가 있습니다. 각 노드는 1부터 n까지 번호가 적혀있습니다. 1번 노드에서 가장 멀리 떨어진 노드의 갯수를 구하려고 합니다. 가장 멀리 떨어진 노드란 최단경로로 이동했을 때 간선의 개수가 가장 많은 노드들을 의미합니다.
- 노드의 개수 n, 간선에 대한 정보가 담긴 2차원 배열 vertex가 매개변수로 주어질 때, 1번 노드로부터 가장 멀리 떨어진 노드가 몇 개인지를 return 하도록 solution 함수를 작성해주세요.
입출력
N |
VERTEX |
RETURN |
6 |
[[3, 6], [4, 3], [3, 2], [1, 3], [1, 2], [2, 4], [5, 2]] |
3 |
접근법
- 그래프의 순회 문제이며, 그래프의 특정점 ( 문제에서는 1로 정해짐. ) 으로 부터 가장 멀리 떨어져 있는 점의 개수를 묻는 문제이다.
- 그래프의 특정 점에서 가장 멀리 떨어져 있는 점의 개수는 BFS( 너비 우선 탐색 ) 으로 구할 수 있으며, visited를 set 형태로 만들어주어, 방문한 적이 있는 지 체크하였다.
- BFS 형식으로 순회를 돌때, Queue 를 이용하여, 해당점으로 부터의 근접점 들을 Queue 에 다 넣어주고, 그 근접점으로부터 파생된 점들을 다시 Queue 에 넣어주어, 순회해 주었다.
#include <string>
#include <vector>
#include <queue>
#include <iostream>
#include <set>
using namespace std;
int solution(int n, vector<vector<int>> edge) {
vector<int> queue_size;
queue<int> queue;
set<int> visited;
visited.insert(1);
queue.push(1);
while (!queue.empty()) {
queue_size.push_back(queue.size());
size_t q_size = queue.size();
for (int i = 0 ; i < q_size; i++) {
int q_front = queue.front();
queue.pop();
for(auto& e: edge) {
if(e[0] == q_front && visited.find(e[1]) == visited.end()) {
queue.push(e[1]);
visited.insert(e[1]);
}
else if (e[1] == q_front && visited.find(e[0]) == visited.end()){
queue.push(e[0]);
visited.insert(e[0]);
}
}
}
}
return queue_size.back();
}