#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; }
Preview:
downloadDownload PNG
downloadDownload JPEG
downloadDownload SVG
Tip: You can change the style, width & colours of the snippet with the inspect tool before clicking Download!
Click to optimize width for Twitter