# 유형 : 그래프 탐색, 최단경로, 플로이드 와샬

# 문제를 잘 보면 그래프에서 모든 정점 사이의 최단 경로를 구하는 문제라는 것을 알 수 있다. 이 때 떠오르는 알고리즘은 플로이드 와샬 알고리즘이다. 플로이드 와샬은 거쳐가는 정점을 기준으로 알고리즘을 수행하는 특징을 가지고 있고, 다이나믹 프로그래밍 기반이다. BFS로도 풀 수 있을 것 같지만 우선 플로이드 와샬로 문제를 해결해보았다. 시간복잡도는 O(V^3)이다.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package bj;
 
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class p1389 {
    static int N,M;
    static int arr[][];
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        arr = new int[N][N];
        for(int i=0; i<N; i++)
            for(int j=0; j<N; j++)
                arr[i][j] = 999999;
        
        for(int i=0; i<M; i++) {
            st = new StringTokenizer(br.readLine());
            int u = Integer.parseInt(st.nextToken());
            int v = Integer.parseInt(st.nextToken());
            arr[u-1][v-1= 1;
            arr[v-1][u-1= 1;
        }
        floyd_warshall();
        
        int min = 999999;
        int index = 0;
        for(int i=0; i<N; i++) {
            int sum = 0;
            for(int j=0; j<N; j++) {
                sum += arr[i][j];
            }
            if(min > sum) {
                min = sum;
                index = i;
            }
//            System.out.println(sum);
        }
        System.out.println(index+1);
    }
    
    public static void floyd_warshall() {
        for(int k=0; k<N; k++) {
            for(int i=0; i<N; i++) {
                for(int j=0; j<N; j++) {
                    arr[i][j] = Math.min(arr[i][j], arr[i][k]+arr[k][j]);
                }
            }
        }
    }
}
 
cs

+ Recent posts