[알고리즘] 백준 11403 - 경로찾기

1. 문제

https://www.acmicpc.net/problem/11403
가중치 없는 방향 그래프 G가 주어졌을 때, 모든 정점 (i, j)에 대해서, i에서 j로 가는 경로가 있는지 없는지 구하는 프로그램을 작성하시오.

2. 입력

첫째 줄에 정점의 개수 N (1 ≤ N ≤ 100)이 주어진다. 둘째 줄부터 N개 줄에는 그래프의 인접 행렬이 주어진다. i번째 줄의 j번째 숫자가 1인 경우에는 i에서 j로 가는 간선이 존재한다는 뜻이고, 0인 경우는 없다는 뜻이다. i번째 줄의 i번째 숫자는 항상 0이다.

3. 풀이

플로이드 워셜 알고리즘을 이용해 풀 수 있다.
0으로 들어오는 입력값은 INF(최대값)으로 바꿔주고, 나머지는 플로이드와 동일하게 풀면 된다.

4. 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class findWay_11403 {
    static int n;
    static int[][] graph;
    static int INF = (int)1e9;

    public static void solution() throws IOException{
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        n = Integer.parseInt(bf.readLine());

        graph = new int[n][n];

		//입력 받기
        StringTokenizer st;
        for(int i=0;i<n;i++){
            st = new StringTokenizer(bf.readLine());
            for(int j=0;j<n;j++){
                int tmp = Integer.parseInt(st.nextToken());
                if(tmp ==0){
                    graph[i][j] = INF;
                }
                else if(tmp==1){
                    graph[i][j] = tmp;
                }
            }
        }

		//그래프 경로 갱신
        for(int k=0;k<n;k++){
            for(int a=0;a<n;a++){
                for(int b=0;b<n;b++){
                    graph[a][b] = Math.min(graph[a][b], graph[a][k]+graph[k][b]);
                }
            }
        }

        //수행된 결과 출력
        for(int a =0; a<n;a++){
            for(int b=0;b<n;b++){
                int tmp = graph[a][b]==INF ? 0 : 1;
                if(b!=n-1){
                    System.out.print(tmp+" ");
                }
                else{
                    System.out.print(tmp);
                }
            }
            System.out.println();
        }
    }
}