Dijkstra算法-lanqiao1122

#include <bits/stdc++.h>
using namespace std;
const long long INF = 0x3f3f3f3f3f3f3f3fLL;
const int N = 3e5 + 5;
struct edge{
    int from, to;//边：起点，终点，权值；起点from没有用到，e[i]的i就是from
    long long w;
    edge(int a,int b,long long c){
        from = a;
        to = b;
        w = c;
    }
};
vector<edge> e[N];//存储图 e数组中存储了N个 vector<edge> 类型的元素
struct node{//点
    int id;
    long long n_dis;//id: 节点，n_dis: 这个点到起点的距离
    node(int b,long long c){
        id = b;
        n_dis = c;
    }
    bool operator < (const node & a)const{//优先级最高的放在priority_queue的顶部，所有重载使得n_dis越小，优先级越高
        return n_dis > a.n_dis;
    }
};
int n, m;
int pre[N];//记录前驱节点
void print_path(int s,int t){//打印起点到t点的最短路径
    if(s == t){
        printf("%d ", s);
        return;
    }
    print_path(s, pre[t]);//先打印到前一个点的路径
    printf("%d ", t);
}
long long dis[N];//记录所有节点到起点的距离
bool done[N];   //done[i] = true 表示节点i的最短路径已经找到
void dijkstra(){
    int s = 1;
    for (int i = 1; i <= n; i++){//初始化
        dis[i] = INF;
        done[i] = false;
    }
    dis[s] = 0;//起点到自己的距离显然是0
    priority_queue<node> Q;//优先队列，存储节点信息
    Q.push(node(s, dis[s]));
    while(!Q.empty()){
        node u = Q.top();   //这里弹出的是距离起点s距离最小的节点
        Q.pop();
        if(done[u.id])
            continue;//如果已经找到了这个节点的最短路径，则跳过该节点
        done[u.id] = true;
        for (int i = 0; i < e[u.id].size(); i++){//检查节点u的所有邻居
            edge y = e[u.id][i];//u.id的第i个邻居是y.to
            if(done[y.to]){
                continue;
            }//丢弃已经找到最短路径的邻居节点
            if(dis[y.to] > y.w + u.n_dis){
                dis[y.to] = y.w + u.n_dis;
                Q.push(node(y.to, dis[y.to]));//拓展新邻居，放入优先队列
                pre[y.to] = u.id;//记录前驱节点
            }
    }
}
}

int main(){
    cin >> n >> m;
    for (int i = 1; i <= n; i++){
        e[i].clear();
    }
    while(m--){
        int u, v, w;
        cin >> u >> v >> w;
        e[u].push_back(edge(u, v, w));
    }
    dijkstra();
    for (int i = 1; i <= n; i++){
        if(dis[i] >= INF)
            cout << "-1 ";
        else{
            cout << dis[i] << " ";
        }
    }
    return 0;
}