문제 & 링크
https://www.acmicpc.net/problem/1916
풀이
1. 정점(도시)의 개수 n과 간선(버스)의 개수 m을 입력받는다.
2. 그래프를 인접 리스트의 형태로 저장한다.
3. 최단 거리 배열인 cost를 큰 값으로 초기화한다.
4. 다익스트라 알고리즘을 사용하여 최단 거리를 탐색한다.
5. 우선 순위 큐를 사용해서 최소 비용부터 탐색한다.
6. 모든 인접 정점을 처리하며 최단거리를 갱신한다.
코드
#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
int cost[1001];
vector<pair<int, int>> V[1001];
void dijkstra(int a) {
priority_queue<pair<int, int>, vector<pair<int,int>>, greater<pair<int, int>>> pq;
pq.push(make_pair(0, a));
cost[a] = 0;
while (!pq.empty())
{
int c = pq.top().first;
int x = pq.top().second;
pq.pop();
if (cost[x] < c)
continue;
for (int i = 0; i < V[x].size(); i++){
int nx = V[x][i].first;
int nc = c + V[x][i].second;
if (cost[nx] > nc){
pq.push(make_pair(nc, nx));
cost[nx] = nc;
}
}
}
}
int main(){
int n, m;
cin >> n >> m;
V[0].push_back(make_pair(0, 0));
memset(cost, 98765432, sizeof(cost));
for (int i = 0; i < m; i++){
int a, b, c;
cin >> a >> b >> c;
V[a].push_back(make_pair(b, c));
}
int st, dt;
cin >> st >> dt;
dijkstra(st);
cout << cost[dt];
}
'Koala - 16기 > 코딩테스트 심화 스터디' 카테고리의 다른 글
[백준/Python] 11047번: 동전 0 (0) | 2024.11.24 |
---|---|
[백준/Python] 13549번: 숨바꼭질 3 (0) | 2024.11.17 |
[백준][Java] 2178 미로 탐색 (0) | 2024.11.10 |
[python] 백준 4963 (0) | 2024.11.10 |
[백준/Python] 2606번: 바이러스 (0) | 2024.11.09 |