PTA 单源最短路径

请编写程序求给定正权有向图的单源最短路径长度。图中包含n个顶点，编号为0至n-1，以顶点0作为源点。

输入格式:

输入第一行为两个正整数n和e，分别表示图的顶点数和边数，其中n不超过20000，e不超过1000。接下来e行表示每条边的信息，每行为3个非负整数a、b、c，其中a和b表示该边的端点编号，c表示权值。各边并非按端点编号顺序排列。

输出格式:

输出为一行整数，为按顶点编号顺序排列的源点0到各顶点的最短路径长度（不含源点到源点），每个整数后一个空格。如源点到某顶点无最短路径，则不输出该条路径长度。

输入样例:

4 4
0 1 1
0 3 1
1 3 1
2 0 1

输出样例:

1 1 

 最基本的Dijkstra算法

#include<bits/stdc++.h>
using namespace std;
const int N = 20005;
struct edge{
    int to, w;
    edge(int a, int b){to = a, w = b;}
};
vector<edge> g[N];
struct node{
    int id, dis;
    node(int a, int b){id = a, dis = b;}
    bool operator < (const node& u) const {return dis > u.dis;}
};
int n, e, done[N], dist[N];
void dijkstra(int s);
int main(){
    scanf("%d%d", &n, &e);
    while(e--){
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        g[a].emplace_back(edge(b, c));
    }
    dijkstra(0);
    for(int i = 1; i < n; ++i){
        if(done[i]) printf("%d ", dist[i]);
    }
    return 0;
}
void dijkstra(int s){
    memset(dist, 0x3f, sizeof dist);
    memset(done, 0, sizeof done);
    priority_queue<node> q;
    dist[s] = 0;
    q.emplace(node(s, dist[s]));
    while(!q.empty()){
        node u = q.top(); q.pop();
        if(done[u.id]) continue;
        done[u.id] = 1;
        for(int i = 0; i < g[u.id].size(); ++i){
            if(!done[g[u.id][i].to] && dist[g[u.id][i].to] > dist[u.id] + g[u.id][i].w){
                dist[g[u.id][i].to] = dist[u.id] + g[u.id][i].w;
                q.emplace(node(g[u.id][i].to, dist[g[u.id][i].to]));
            }
        }
    }
}