dijkstra new ver 1
Wed Sep 04 2024 11:27:07 GMT+0000 (Coordinated Universal Time)
Saved by
@LizzyTheCatto
#include <bits/stdc++.h>
using namespace std;
void D(int N, vector<pair<int, int>> adj[], int source) {
vector<int> dist(N, 1000000); // Khởi tạo khoảng cách đến tất cả các đỉnh là vô cùng
dist[source] = 0; // Khoảng cách từ đỉnh nguồn đến chính nó là 0
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq; // Hàng đợi ưu tiên
pq.push({0, source});
while (!pq.empty()) { // Trong khi hàng đợi không rỗng
int u = pq.top().second; // Lấy đỉnh có khoảng cách nhỏ nhất
int d = pq.top().first; // Khoảng cách từ nguồn đến đỉnh u
pq.pop();
// Duyệt các đỉnh kề của đỉnh u
for (int i = 0; i < adj[u].size(); i++) {
int v = adj[u][i].first; // Đỉnh kề
int weight = adj[u][i].second; // Trọng số của cạnh
// Nếu tìm được đường đi ngắn hơn đến đỉnh v
if (dist[v] > dist[u] + weight) {
dist[v] = dist[u] + weight;
pq.push({dist[v], v}); // Đẩy khoảng cách mới vào hàng đợi
}
}
}
// In ra kết quả khoảng cách từ đỉnh nguồn đến tất cả các đỉnh khác
for (int i = 0; i < N; i++) {
cout << "Khoảng cách từ " << source << " đến " << i << " là " << dist[i] << endl;
}
}
int main() {
int N, M; // Số đỉnh, số cạnh
cin >> N >> M;
vector<pair<int, int>> adj[N]; // Mảng vector để lưu đồ thị
for (int i = 0; i < M; i++) {
int a, b, c;
cin >> a >> b >> c;
adj[a].push_back({b, c}); // Đồ thị có hướng
adj[b].push_back({a, c}); // Nếu đồ thị là vô hướng
}
int source;
cin >> source;
D(N, adj, source);
return 0;
}
content_copyCOPY
https://www.programiz.com/cpp-programming/online-compiler/
Comments