PS/BOJ
[알고리즘] 백준 1613번: 역사 (C++)
도비(Doby)
2021. 12. 6. 15:31
https://www.acmicpc.net/problem/1613
1613번: 역사
첫째 줄에 첫 줄에 사건의 개수 n(400 이하의 자연수)과 알고 있는 사건의 전후 관계의 개수 k(50,000 이하의 자연수)가 주어진다. 다음 k줄에는 전후 관계를 알고 있는 두 사건의 번호가 주어진다.
www.acmicpc.net
또 똑같은 키워드다.
(https://draw-code-boy.tistory.com/151)
(https://draw-code-boy.tistory.com/153)
다만, 시간 초과가 나와서
ios_base::sync_with_stdio(false);
cin.tie(NULL);
를 해주고,
값을 복사해주는 cache의 역할이 사실상 의미가 없기 때문에
graph에 입력받은 값을 복사하지 않고, 즉각적으로 cache에 입력을 받고, 플로이드 와샬을 돌렸다.
[키워드]
플로이드 와샬로 각 노드 간의 전후 관계(?), 관련 있는 노드가 몇 개인가 등 여러 문제를 풀 수 있는 것을 알아냈다.
[AC 코드]
#include <iostream>
#include <cmath>
#define MAX 500 + 1
#define INF 987654321
using namespace std;
int cache[MAX][MAX];
int n, m;
void floydWarshall() {
for (int k = 1; k <= n; k++) {
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
if (cache[i][k] != INF && cache[k][j] != INF) {
cache[i][j] = min(cache[i][j], cache[i][k] + cache[k][j]);
}
}
}
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
if (i == j) cache[i][j] = 0;
}
}
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cin >> n >> m;
// init
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
cache[i][j] = INF;
}
}
for (int i = 0; i < m; i++) {
int a, b;
cin >> a >> b;
cache[a][b] = 1;
}
floydWarshall();
int s;
cin >> s;
for (int i = 0; i < s; i++) {
int a, b;
cin >> a >> b;
if (cache[a][b] != 0 && cache[a][b] != INF) {
cout << -1 << '\n';
}
else if (cache[b][a] != 0 && cache[b][a] != INF) {
cout << 1 << '\n';
}
else {
cout << 0 << '\n';
}
}
return 0;
}