Algorithm/BOJ
[C/C++] 백준 14284번 간선 이어가기 2
poopooreum
2024. 4. 3. 21:00
반응형
✏️ 문제 링크
https://www.acmicpc.net/problem/14284
14284번: 간선 이어가기 2
정점 n개, 0개의 간선으로 이루어진 무방향 그래프가 주어진다. 그리고 m개의 가중치 간선의 정보가 있는 간선리스트가 주어진다. 간선리스트에 있는 간선 하나씩 그래프에 추가해 나갈 것이다.
www.acmicpc.net
✏️ 문제 설명
✏️ 문제 풀이
단순한 다익스트라 알고리즘 구현 문제입니다
풀이는 아래 코드 참조
✏️ 문제 코드
#include<iostream>
#include<vector>
#include<limits.h>
#include<queue>
using namespace std;
typedef pair<int, int>cur;
vector<vector<cur>>tree;
vector<bool>visit;
vector<int>dist;
int main() {
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
int N, M;
cin >> N >> M;
tree.resize(N + 1);
visit.resize(N + 1);
dist.resize(N + 1);
fill(visit.begin(), visit.end(), false);
fill(dist.begin(), dist.end(), INT_MAX);
for (int i = 0; i < M; i++) {
int start, end, value;
cin >> start >> end >> value;
tree[start].push_back(cur(end, value));
tree[end].push_back(cur(start, value));
}
int s, t;
cin >> s >> t;
priority_queue<cur, vector<cur>, greater<cur>>pq;
pq.push(cur(0, s));
dist[s] = 0;
while (!pq.empty()) {
cur temp = pq.top();
int node = temp.second;
pq.pop();
if (!visit[node]) {
visit[node] = true;
for (cur f : tree[node]) {
dist[f.first] = min(dist[f.first], dist[node] + f.second);
pq.push(cur(dist[f.first], f.first));
}
}
}
cout << dist[t];
}
반응형