인프런 커뮤니티 질문&답변

• 학습 관련 질문을 남겨주세요. 구체적으로 적을수록 좋아요!

• 마크다운과 단축키를 활용하면 글을 더 편하게 작성할 수 있어요.

• 커뮤니티 질문 & 답변에 비슷한 내용이 있었는지 먼저 검색해보세요.

편하게 질문주세요:) 직접 답변 드리도록 하겠습니다~

정답 풀이 코드가 이상한거 같아요 ,,!

1. e1[2] - e2[2] 로 코드를 구성하면 OutOfBound에러가 발생해요~

   
   e1[0] - e2[0] 이렇게 구성해야하는게 아닌가요 ?

    

2. curCost + edge[0] 에서 edge[0]가중치가 아니라 출발 노드 이므로edge[2] 을 더 하는게 맞지 않나요? curCost + edge[2]

    

3. new int[]{curNode, nextCost}
   큐에 {curNode, nextCost} 추가 → 배열 형식이 {cost, node}여야 하는데 순서와 값이 모두 잘못된거같은데 맞을까요 ?
   
   

   public class NetworkDelayTime {
     public int solution(int[][] times, int n, int k) {
       // 그래프 구현
       Map<Integer, List<int[]>> graph = Arrays.stream(times)
       .collect(Collectors.groupingBy(t -> t[0]));
   
       // 다익스트라 알고리즘
       Map<Integer, Integer> costs = new HashMap<>();
       Queue<int[]> pq = new PriorityQueue<>((e1, e2) -> e1[0] - e2[0]);
       pq.add(new int[]{0, k});
   
       while (!pq.isEmpty()) {
         int[] cur = pq.remove();
         int curCost = cur[0];
         int curNode = cur[1];
         if (!costs.containsKey(curNode)) {
           costs.put(curNode, curCost);
           if (!graph.containsKey(curNode)) continue;
           for (int[] edge : graph.get(curNode)) {
             int nextCost = curCost + edge[2];
             pq.add(new int[]{nextCost, edge[1]});
           }
         }
       }
       // 방문 못한 노드 찾기
       for (int node = 1; node < n + 1; node++) {
         if (!costs.containsKey(node)) {
           return -1;
         }
       }
       // 최소값중에서 최대값 구하기
       return costs.values().stream()
       .mapToInt(Integer::intValue).max().getAsInt();
     }
   
     public static void main(String[] args) {
       NetworkDelayTime solution = new NetworkDelayTime();
       int[][] times = {
         {2, 1, 2},
         {2, 3, 5},
         {2, 4, 1},
         {4, 3, 3}
       };
       int result = solution.solution(times, 4, 2);
       System.out.println(result);
     }
   }

   이렇게 구현하니 제대로 돌아가서 문의드립니다