[백준] 자바 11403 경로 찾기

문제

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

입력

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

출력

총 N개의 줄에 걸쳐서 문제의 정답을 인접 행렬 형식으로 출력한다. 정점 i에서 j로 가는 경로가 있으면 i번째 줄의 j번째 숫자를 1로, 없으면 0으로 출력해야 한다.

---

테스트 케이스

 

입력 1

7
0 0 0 1 0 0 0
0 0 0 0 0 0 1
0 0 0 0 0 0 0
0 0 0 0 1 1 0
1 0 0 0 0 0 0
0 0 0 0 0 0 1
0 0 1 0 0 0 0

출력 1

1 0 1 1 1 1 1 
0 0 1 0 0 0 1 
0 0 0 0 0 0 0 
1 0 1 1 1 1 1 
1 0 1 1 1 1 1 
0 0 1 0 0 0 1 
0 0 1 0 0 0 0 

 

입력 2

3
0 1 0
0 0 1
1 0 0

출력 2

1 1 1 
1 1 1 
1 1 1 

 

입력 3

4
1 0 1 0
0 1 0 1
1 0 1 0
0 1 0 1

출력 3

1 0 1 0 
0 1 0 1 
1 0 1 0 
0 1 0 1 

---

접근

플로이드 워샬 문제입니다.

플로이드 워샬은 모든 점 간에 최단거리를 구할 때, 사용하는 알고리즘인데요.

이를 문제에 적용하니 쉽게 해결되었습니다.

삼중 포문을 이용하여 3가지 점을 돌면서(시작 점, 도착 점, 거쳐 가는 점) 1을 발견하게 되면 해당 배열에 1을 넣어주면서 정답 배열을 완성시킵니다.

(자세한 건 코드를 보면 이해가 되실 겁니다.)

---

코드

	import java.util.*;
import java.util.stream.IntStream;
import java.io.*;
	
	class Main {
		
		public static void main(String args[]) throws IOException {
	    	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
	    	BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

	    	int N = Integer.parseInt(br.readLine());
	    	int map[][] = new int[N][N];
	    	
	    	for(int i=0; i<N; i++) {
	    		StringTokenizer st = new StringTokenizer(br.readLine());
	    		for(int j=0; j<N; j++) {
	    			map[i][j] = Integer.parseInt(st.nextToken());
	    		}
	    	}
	    	//점 k를 지나간다.
	    	for(int k=0; k<N; k++) {
	    		//점 i에서 시작
	    		for(int i=0; i<N; i++) {
	    			//점 j에서 시작
	    			for(int j=0; j<N; j++) {
	    				if(map[i][k] == 1 && map[k][j] == 1)
	    					map[i][j] = 1;
	    			}
	    		}
	    	}
	    	for(int i=0; i<N; i++) {
	    		for(int j=0; j<N; j++) {
	    			bw.write(String.valueOf(map[i][j]+" "));
	    		}
	    		bw.write("\n");
	    	}
	    		
	    	br.close();
	    	bw.flush();
	    	bw.close();
	    }
	}

---

주의

정답 배열 사이사이에 띄어쓰기 있습니다.