코딩/백준

[백준]1238번 파티 - C/C++

최선을 다하는 2022. 2. 28. 10:55

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

 

1238번: 파티

첫째 줄에 N(1 ≤ N ≤ 1,000), M(1 ≤ M ≤ 10,000), X가 공백으로 구분되어 입력된다. 두 번째 줄부터 M+1번째 줄까지 i번째 도로의 시작점, 끝점, 그리고 이 도로를 지나는데 필요한 소요시간 Ti가 들어

www.acmicpc.net

문제

N개의 숫자로 구분된 각각의 마을에 한 명의 학생이 살고 있다.

어느 날 이 N명의 학생이 X (1 ≤ X ≤ N)번 마을에 모여서 파티를 벌이기로 했다. 이 마을 사이에는 총 M개의 단방향 도로들이 있고 i번째 길을 지나는데 Ti(1 ≤ Ti ≤ 100)의 시간을 소비한다.

각각의 학생들은 파티에 참석하기 위해 걸어가서 다시 그들의 마을로 돌아와야 한다. 하지만 이 학생들은 워낙 게을러서 최단 시간에 오고 가기를 원한다.

이 도로들은 단방향이기 때문에 아마 그들이 오고 가는 길이 다를지도 모른다. N명의 학생들 중 오고 가는데 가장 많은 시간을 소비하는 학생은 누구일지 구하여라.

입력

첫째 줄에 N(1 ≤ N ≤ 1,000), M(1 ≤ M ≤ 10,000), X가 공백으로 구분되어 입력된다. 두 번째 줄부터 M+1번째 줄까지 i번째 도로의 시작점, 끝점, 그리고 이 도로를 지나는데 필요한 소요시간 Ti가 들어온다. 시작점과 끝점이 같은 도로는 없으며, 시작점과 한 도시 A에서 다른 도시 B로 가는 도로의 개수는 최대 1개이다.

모든 학생들은 집에서 X에 갈수 있고, X에서 집으로 돌아올 수 있는 데이터만 입력으로 주어진다.

출력

첫 번째 줄에 N명의 학생들 중 오고 가는데 가장 오래 걸리는 학생의 소요시간을 출력한다.


 

모든 점에서 한점으로 갔다가 한 점에서 모든 점으로 나가는 비용을 구하는 문제여서 처음에는 크루스칼 + 다익스트라로 풀어야 하나 생각을 했지만 플로이드는 모든 점에서 모든 점으로 가는 비용을 알 수 있으니 단순히 플로이드-워샬 알고리즘을 사용하여 답을 도출해낼 수 있을 것 같았다. 모든 점에서 점으로의 최소비용을 더한 다음에 가는 길 + 오는 길의 최댓값만 구한다면 쉽게 풀 수 있는 문제이다.

#include <iostream>
#include <algorithm>
#define MAX 987654321
using namespace std;

int N, M, X,a,b,c;
int arr[1010][1010];

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

	cin >> N >> M >> X;
	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) 
			arr[i][j] = MAX;
		arr[i][i] = 0;
	}
	for (int i = 1; i <= M; i++) {
		cin >> a >> b >> c;
		arr[a][b] = c;
	}
	for (int k = 1; k <= N; k++) {
		for (int i = 1; i <= N; i++) {
			for (int j = 1; j <= N; j++) {
				if (arr[i][j] > arr[i][k] + arr[k][j])
					arr[i][j] = arr[i][k] + arr[k][j];
			}
		}
	}
	int res = 0;
	for (int i = 1; i <= N; i++) {
		res = max(res, arr[i][X] + arr[X][i]);
	}
	cout << res;

}

플로이드-워샬을 풀면 항상 초기화를 하지 않고 돌렸다가 다른 값이 나오곤 했는데 이번에도 초기화 안 함 -> memset으로 초기화의 과정으로 두 번이나 이상한 짓을 해버리고 말았다!

방학이 이틀 남았다. 슬슬 혼자 공부하는 것이 질려가던 시기에 개강하니깐 좋은 것 같다. 저번학기는 가볍게 4 전공을 했지만 이번에는 6 전공에 프로젝트 과목도 많다. 힘들지만 정신만 차리고 있으면 보람찬 학기가 될 것 같다!