[프로그래머스 lv3] 가장 먼 노드 (다익스트라, 코틀린)

https://github.com/mangbaam/CodingTest

GitHub - mangbaam/CodingTest: 프로그래머스, 백준 등 코딩테스트 풀이를 기록하는 저장소입니다.

밑의 사진을 클릭하면 문제 링크로 이동합니다

 

 

 

문제

문제 설명

n개의 노드가 있는 그래프가 있습니다. 각 노드는 1부터 n까지 번호가 적혀있습니다. 1번 노드에서 가장 멀리 떨어진 노드의 갯수를 구하려고 합니다. 가장 멀리 떨어진 노드란 최단경로로 이동했을 때 간선의 개수가 가장 많은 노드들을 의미합니다.

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

제한사항

• 노드의 개수 n은 2 이상 20,000 이하입니다.
• 간선은 양방향이며 총 1개 이상 50,000개 이하의 간선이 있습니다.
• vertex 배열 각 행 [a, b]는 a번 노드와 b번 노드 사이에 간선이 있다는 의미입니다.

예제

입출력 예 설명

예제의 그래프를 표현하면 아래 그림과 같고, 1번 노드에서 가장 멀리 떨어진 노드는 4,5,6번 노드입니다.

---

나의 풀이

이 문제는 특정 위치(1번 노드)에서 나머지 노드까지의 거리를 구해야 해서 다익스트라 최단 거리 탐색 알고리즘을 사용해서 풀이했다.

다익스트라 최단 거리 탐색 알고리즘에 대해서는 아래 글을 참고하면 될 것 같다.

2022.08.16 - 코틀린 다익스트라 알고리즘 Kotlin dijkstra algorithm

코틀린 다익스트라 알고리즘 Kotlin dijkstra algorithm

 

일반적인 다익스트라 알고리즘에서 시작 점을 1번 노드로 고정하고, 모든 거리를 1로 설정하고 무방향 그래프인 것만 신경 쓰면 나머지는 모두 동일하다.

탐색을 완료한 후 가장 큰 값을 찾아서 그 값의 개수를 세면 된다.

import java.util.*

class Solution {
    val Inf = 55555
    
    fun solution(n: Int, edge: Array<IntArray>): Int {
        var answer = 0
        
        val graph = Array(n + 1) { mutableListOf<Pair<Int, Int>>() }
        val distance = Array(n + 1) { Inf }.apply { this[1] = 0 }
        val q = PriorityQueue<Pair<Int, Int>>() { p1, p2 ->
            when {
                p1.second > p2.second -> 1
                p1.second < p2.second -> -1
                else -> 0
            }
        }
        
        edge.forEach { (a, b) ->
            graph[a].add(Pair(b, 1))
            graph[b].add(Pair(a, 1))
        }
        
        q.add(Pair(1, 0))
        
        while (q.isNotEmpty()) {
            val (now, dist) = q.poll()
            
            if (distance[now] < dist) continue
            
            for (i in graph[now]) {
                val cost = dist + i.second
                if (cost < distance[i.first]) {
                    distance[i.first] = cost
                    q.add(Pair(i.first, cost))
                }
            }
        }

        val maxDist = distance.drop(1).maxOrNull()
        return distance.count { it == maxDist }
    }
}

채점 결과