취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 1507번 문제인 궁금한 민호이다.
문제는 아래 링크를 확인하자.

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

서로 다른 두 노드 r, c 사이의 최단경로 위에 다른 노드 k가 존재한다면, 이 최단경로의 길이는 (r과 k 사이의 최단경로의 길이) + (k와 c 사이의 최단경로의 길이)와 같다는 점을 이용하자.

즉, r과 c 사이의 최단거리와 (r과 k 사이의 최단경로의 길이) + (k와 c 사이의 최단경로의 길이)가 같은 k가 존재한다면 r과 c를 직접 잇는 도로는 존재하지 않아도 된다.

만약 r과 c 사이의 최단거리보다 작은 (r과 k 사이의 최단경로의 길이) + (k와 c 사이의 최단경로의 길이) 값이 존재한다면 주어진 입력이 잘못된 것이므로 -1을 출력해주자.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;

int arr[20][20];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	int N; cin >> N;
	for (int r = 0; r < N; r++) {
		for (int c = 0; c < N; c++) cin >> arr[r][c];
	}
	int ans = 0;
	for (int r = 0; r < N; r++) {
		for (int c = r + 1; c < N; c++) {
			bool chk = 1;
			for (int k = 0; k < N; k++) {
				if (r == k || c == k) continue;
				if (arr[r][k] + arr[k][c] < arr[r][c]) {
					cout << -1;
					return 0;
				}
				if (arr[r][k] + arr[k][c] == arr[r][c]) chk = 0;
			}
			if (chk) ans += arr[r][c];
		}
	}

	cout << ans;
}