[level 3] Title: 가장 먼 노드, Time: 54.44 ms, Memory: 107 MB -BaekjoonHub

Author: chech2

프로그래머스/3/49189. 가장 먼 노드/README.md

@@ -4,7 +4,7 @@
 
 ### 성능 요약
 
-메모리: 108 MB, 시간: 34.23 ms
+메모리: 107 MB, 시간: 54.44 ms
 
 ### 구분
 
@@ -16,7 +16,7 @@
 
 ### 제출 일자
 
-2025년 03월 27일 13:10:28
+2025년 05월 23일 15:50:21
 
 ### 문제 설명
 

프로그래머스/3/49189. 가장 먼 노드/가장 먼 노드.java

@@ -1,7 +1,8 @@
 import java.util.*;
 
 class Solution {
-    List<Integer>[] tree;
+    int n;
+    List<Integer>[] graph;
     boolean[] visited;
     int[] d;
     public int solution(int n, int[][] edge) {
@@ -12,52 +13,54 @@ public int solution(int n, int[][] edge) {
     }
 
     int bfs(){
-        int ans, max;
-        Queue<Integer> q = new ArrayDeque();
-        q.add(1);
+        PriorityQueue<node> q = new PriorityQueue<>((o1, o2) -> o1.d - o2.d);
+        q.add(new node(1, 1));
         visited[1] = true;
+
 
-        int next, now;
+        int next, answer;
+        node now;
+        answer = 0;
         while(!q.isEmpty()){
             now = q.poll();
 
-            for(int i = 0; i < tree[now].size(); i++){
-                next = tree[now].get(i);
-                
+            answer = Math.max(answer, now.d);
+            d[now.d] += 1;
+            for(int i = 0; i < graph[now.n].size(); i++){
+                next = graph[now.n].get(i);
                 if(visited[next]) continue;
-                visited[next] = true;
-                d[next] = d[now] + 1;
 
-                q.add(next);
+                visited[next] = true;
+                q.add(new node(next, now.d + 1));
             }
         }
-        
-        Arrays.sort(d);
-        ans = 0;
-        max = d[d.length - 1];
-        for(int i = d.length - 1; 0 < i; i--){
-            
-            if(d[i] < max) break;
-            ans++;
-        }
-        return ans;
+        return d[answer];
     }
 
     void preSetting(int n, int[][] edge){
-        tree = new ArrayList[n + 1];
-        d = new int[n + 1]; 
+        this.n = n;
+        graph = new ArrayList[n + 1];
         visited = new boolean[n + 1];
-        visited[1] = true;
+        d = new int[n + 1];
 
-        for(int i = 0; i < n + 1; i++) tree[i] = new ArrayList<>();
+        for(int i = 0; i < n + 1; i++) graph[i] = new ArrayList<>();
 
         int a, b;
         for(int i = 0; i < edge.length; i++){
             a = edge[i][0];
             b = edge[i][1];
 
-            tree[a].add(b);
-            tree[b].add(a);            
+            graph[a].add(b);
+            graph[b].add(a);
+        }
+    }
+    
+    class node{
+        int n, d;
+        
+        node(int n, int d){
+            this.n = n;
+            this.d = d;
         }
     }
 }