19-miniron-v

[miniron-v]

🔗 문제 링크

https://www.acmicpc.net/problem/1753
1753번: 최단경로
분류: 그래프, 다익스트라(데이크스트라)

✔️ 소요된 시간

1시간

✨ 수도 코드

다익스트라를 제대로 다루지 못 해서, 블로그를 보며 구현해봤습니다.

https://velog.io/@717lumos/알고리즘-다익스트라Dijkstra-알고리즘
https://suyeon96.tistory.com/31

이 두 블로그를 참고해, 우선순위 큐 방식으로 다익스트라를 구현했습니다.

📚 전체 코드

#include <iostream>
#include <vector>
#include <queue>

const int INF = 3000001;

std::vector<int> dijkstra(int start, int n, std::vector<std::vector<std::pair<int, int>>>& graph) {
	std::vector<int> min_weight(n + 1, INF);
	std::priority_queue<std::pair<int, int>, std::vector<std::pair<int, int>>, std::greater<std::pair<int, int>>> pq;

	min_weight[start] = 0;
	pq.push({ 0, start });

	while (pq.empty() == false) {
		int cur_weight = pq.top().first;
		int cur_node = pq.top().second;
		pq.pop();

		for (auto& pair : graph[cur_node]) {
			int next_node = pair.first;
			int next_weight = cur_weight + pair.second;

			if (next_weight < min_weight[next_node]) {
				min_weight[next_node] = next_weight;
				pq.push({ next_weight, next_node});
			}
		}
	}

	return min_weight;
}

int main() {
	// 입력
	int V, E, start;
	std::cin >> V >> E >> start;

	std::vector<std::vector<std::pair<int, int>>> graph(V + 1);
	for (int i = 0; i < E; ++i) {
		int u, v, w;
		std::cin >> u >> v >> w;

		graph[u].push_back({v, w});
	}
	
	// 계산
	std::vector<int> min_weight = dijkstra(start, V, graph);

	// 출력
	for (int i = 1; i < min_weight.size(); ++i) {
		if (min_weight[i] == INF) {
			std::cout << "INF\n";
			continue;
		}

		std::cout << min_weight[i] << "\n";
	}
}

📚 새롭게 알게된 내용

C++에서 pair를 만들 때 std::make_pair(A, B);를 쓰지 않고 {A, B}만 해도 페어가 만들어진다. (버전 낮아서 안 되는 줄 알았는데 신세계)

추가로 { 가중치, 노드 }로 넣으면 성공, { 노드, 가중치 }로 넣으면 시간 초과가 떴는데, pq에 커스텀 함수를 쓰지 않고 greater<std::pair<int, int>>를 썼기 때문으로 보인다. 내림차순 정렬해야 최솟값이 나가는 걸보면, 우선순위 큐의 pop이 뒤에서 일어나는 듯.

greater<std::pair<int, int>>로 int형처럼 내림차순 정렬이 가능하다. 이때 비교 기준은 first다.

[9kyo-hwang]

std::greater<>

pair<int, int> 타입이 추론되기 때문에 굳이 pair<int, int>를 적어주지 않아도 된답니당

[Redish03]

그럼요~

[miniron-v]

유니티도 new Vector3()로 적던 거 new()로 생성되게 해놨던데... 점점 코드가 짧아지는 쪽으로 발전하네요.

[9kyo-hwang]

if(min_weight[cur_node] < cur_weight) {
    continue;
}

보통 여기에 이런 식으로 인접 노드에 대해 검사할 필요도 없는 경우를 걸러주는 케이스를 넣어주기도 하더라구요. 일부 문제는 이 조건문이 없으면 터지더라구요...

[miniron-v]

아마 코딩 스타일의 차이인 것 같지만,

for (auto& pair : graph[cur_node]) {
	int next_node = pair.first;
	int next_weight = cur_weight + pair.second;

	if (next_weight < min_weight[next_node]) {
		min_weight[next_node] = next_weight;
		pq.push({ next_weight, next_node});
	}
}

이 구문에서 검사할 필요가 없는 노드는 큐에 push 되지 않습니다. 터지지 않는 이유!

[9kyo-hwang]

2021 KAKAO BLIND RECRUITMENT 합승 택시 요금
1504 특정한 최단 경로
1238 파티

개인적으로 괜찮게 생각하는 다익스트라 응용 문제들입니다. 한 번 기회되면 풀어보세요~

[Redish03]

고생하셨습니다 :)

다익스트라의 기본 같은 문제라 저도 복습 하고 갑니다..,ㅎㅎ

[Redish03]

그럼요~

[Redish03]

제 PR에서 교황쿤이 얘기해줬던 거 같은데 make_pair()이나 메서드 보다 {x, y}로 넣는걸 추천드리더라구요 :)

[miniron-v]

너무 편하더라고요... std::make_pair(x, y)를 안 써도 도딘다니

[bomik0221]

저도 다익스트라 문제는 안 해봤는데, 덕분에 알아갑니다! 고생하셨습니다 :)

[2secondag]

코드 잘 봤습니다~!
다익스트라 문제는 처음 보는 것 같은데 신기하네요 배워갑니다...!!