floyd warshall :: 백준 :: 역사 :: 1613

역사

https://www.acmicpc.net/problem/1613

[문제 풀이]

첫 줄에 있는 사건의 개수 n과 전후관계 개수 k을 입력받는다.

map[n][n] 배열을 생성한 후

전후관계 k개(row, col 형태)를 입력받는데

map[row][col] = -1

map[col][row] = 1

이렇게 설정한다.

예제로 입력된 정보를 토대로 맵을 생성해보면 

	-1	-1
1		-1	-1
1	1		-1
	1	1

이런 형태의 맵이 나오게 된다.

이 상태에서 플로이드 와샬 알고리즘을 사용하여 완전탐색을 하면 된다.

if(map[row][k] != 0 && map[row][k] == map[l][col]){

map[row][col] = map[row][k];

}

0	-1	-1	-1	0
1	0	-1	-1	0
1	1	0	-1	0
1	1	1	0	0
0	0	0	0	0

탐색결과 위와 같은 형태로 map이 갱신되고

알고싶은 위치를 읽어서 출력하면 된다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        // TODO Auto-generated method stub
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        StringBuffer sb = new StringBuffer();
        int n = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());
        int[][] map = new int[n + 1][n + 1];
        
        int r, c;
        //맵 생성하는 부분
        for (int i = 0; i < k; i++) {
            st = new StringTokenizer(br.readLine());
            r = Integer.parseInt(st.nextToken());
            c = Integer.parseInt(st.nextToken());
            map[r][c] = -1;
            map[c][r] = 1;
        }
        
        //탐색부분
        for(int l = 1; l <= n; l++){
            for(int row = 1; row <= n; row++){
                for(int col = 1; col <= n; col++){
                    if(map[row][l] != 0 && map[row][l] == map[l][col]){
                        map[row][col] = map[row][l];
                    }
                }
            }
        }
        //출력하는 부분
        st = new StringTokenizer(br.readLine());
        int s = Integer.parseInt(st.nextToken());
        for (int i = 0; i < s; i++) {
            st = new StringTokenizer(br.readLine());
            sb.append(map[Integer.parseInt(st.nextToken())][Integer.parseInt(st.nextToken())] + "\n");
        }
        System.out.println(sb.toString());
    }
}