[BOJ] 1507. 궁금한 민호

플로이드 와샬을 활요하는 문제였다.

이 문제같은 경우 잘생각할 필요가 있는 문제이다. 이미 주어지는 입력이 최소 이동 시간이고, 여기서 구해야할 것이 필요없는 도로를 골라내는 문제이다. 즉 입력을 이용하여 다시 플로이드와샬을 돌면서 map[i][j] = map[i][k] + map[k][j] 인 경우 map[i][j] 경로가 필요없게 됨으로 이것을 찾아주었다. 또한 불가능한 경우는 최소값을 갱신할 수 있을경우이기에 이것에 대한 처리도 같이 해주었다.

package algostudy2;
 
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class Main_bj_1507_궁금한민호 {
    static int[][] map;
    static int[][] check;
    static int N;
 
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;
 
        N = Integer.parseInt(br.readLine());
 
        map = new int[N + 1][N + 1];
        check = new int[N + 1][N + 1];
 
        for (int i = 1; i <= N; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 1; j <= N; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }
 
        boolean result = floyd();
 
        if (result) {
            int sum = 0;
            for (int i = 1; i <= N; i++) {
                for (int j = 1; j <= N; j++) {
                    if (check[i][j] != -1) {
                        sum += map[i][j];
                    }
                }
            }
            System.out.println(sum / 2);
        } else {
            System.out.println(-1);
        }
 
    }
 
    static boolean floyd() {
        for (int k = 1; k <= N; k++) {
            for (int i = 1; i <= N; i++) {
                for (int j = 1; j <= N; j++) {
                    if (i != j && i != k && j != k) {
                        if (map[i][j] == map[i][k] + map[k][j]) {
                            check[i][j] = -1;
                        } else if (map[i][j] > map[i][k] + map[k][j]) {
                            return false;
                        }
                    }
                }
            }
        }
 
        return true;
    }
 
}