728x90


문제풀이

- 이 문제는 플로이드 와샬(Floyd Warshall) 알고리즘 문제이다.

플로이드 와샬 알고리즘은 모든 정점에서 모든 정점으로의 거리의 최소값을 구할 수 있다.

a 정점에서 b 정점으로 이동 하는 배열을 입력받아서

a -> k -> b 로 가는 경로가 존재하는지 확인하며 값을 채워 넣는다.

시간 복잡도는 O(n^3) 이 걸리는 방법이다.

 

https://www.youtube.com/watch?v=9574GHxCbKc 

이 영상을 참고하여 공부하였다.

대학때 알고리즘 수업을 들었었는데, 대학수업 보다 요즘 코딩 유튜버들이 훨씬 설명을 잘한다고 느낀다.

 


소스코드

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
import java.util.*;
import java.io.*;
class Main{
    static int n;
    static int[][] arr;
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;
        n = Integer.parseInt(br.readLine());
        arr = new int[n][n];
        for(int i=0;i<n;i++)
        {
            st = new StringTokenizer(br.readLine());
            for(int j=0;j<n;j++)
            {
                arr[i][j] = Integer.parseInt(st.nextToken());
            }
        }
        
        //Floyd Warshall
        for(int k=0;k<n;k++)
        {
            for(int i=0;i<n;i++)
            {
                for(int j=0;j<n;j++)
                {
                    if(arr[i][k]==1 && arr[k][j]==1)
                    {
                        arr[i][j] = 1;
                    }
                }
            }
        }
        
        
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                sb.append(arr[i][j]).append(' ');
            }
            sb.append('\n');
        }
        System.out.print(sb);
    }
}
cs
LIST

+ Recent posts