728x90
DFS나 BFS로 풀려다가 코드 짜기가 너무 복잡할 것 같아서 간단하게 플로이드로 풀었다.
#include <iostream>
#include <tuple>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
cin.tie(NULL);
ios_base::sync_with_stdio(false);
int n;
cin >> n;
const int INF = 1000;
vector<vector<int>> d(n, vector<int>(n, INF));
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
int temp;
cin >> temp;
if (temp != 0) {
d[i][j] = temp;
}
}
}
for (int k = 0; k < n; k++) {
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
}
}
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (d[i][j] == INF) cout << 0 << " ";
else cout << 1 << " ";
}
cout << endl;
}
return 0;
}
728x90
'알고리즘 문제' 카테고리의 다른 글
[백준] 3613번 Java vs C++ (0) | 2020.02.08 |
---|---|
[백준] 10808번 알파벳 개수 (0) | 2020.02.08 |
[백준] 11404번 플로이드 (0) | 2020.02.07 |
[백준] 1977번 완전제곱수 (0) | 2020.02.07 |
[백준] 2163번 초콜릿 자르기 (0) | 2020.02.07 |