water

11  •  3个月前


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

using namespace std;

struct Edge {
    int to, weight;
};

int main() {
    int n, m, s;
    cin >> n >> m >> s;
    
    // 使用邻接表存储图
    vector<vector<Edge>> graph(n + 1);
    
    for (int i = 0; i < m; ++i) {
        int u, v, w;
        cin >> u >> v >> w;
        graph[u].push_back({v, w});
    }
    
    // 距离数组,初始化为无穷大
    vector<int> dist(n + 1, INT_MAX);
    dist[s] = 0;
    
    // 最小堆,存储{距离, 节点}
    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
    pq.push({0, s});
    
    while (!pq.empty()) {
        int d = pq.top().first;
        int u = pq.top().second;
        pq.pop();
        
        // 如果当前距离大于已经计算出的最短距离,跳过
        if (d > dist[u]) continue;
        
        // 遍历所有邻接边
        for (const auto& edge : graph[u]) {
            int v = edge.to;
            int weight = edge.weight;
            
            // 如果找到更短路径,更新并加入优先队列
            if (dist[u] + weight < dist[v]) {
                dist[v] = dist[u] + weight;
                pq.push({dist[v], v});
            }
        }
    }
    
    // 输出结果
    for (int i = 1; i <= n; ++i) {
        if (dist[i] == INT_MAX) {
            cout << -1 << " ";
        } else {
            cout << dist[i] << " ";
        }
    }
    cout << endl;
    
    return 0;
}

评论:

请先登录,才能进行评论