单源最短路径

请编写程序求给定正权有向图的单源最短路径长度。图中包含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 

代码长度限制

16 KB

时间限制

100 ms

内存限制

20 MB

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 5;
const int inf = 0x3f3f3f3f;
typedef pair<int, int> PII;
int h[N], e[N], ne[N], idx;
int w[N];
int dist[N];
bool st[N];
int n, m;
void add(int x, int y, int c)
{
    w[idx]=c, e[idx]=y, ne[idx]=h[x], h[x]=idx++;
}
void dij()
{
    memset(dist, 0x3f, sizeof dist);
    dist[0]=0;
    priority_queue<PII, vector<PII>, greater<PII> >heap;
    heap.push({0, 0});
    while(heap.size())
    {
        PII k = heap.top();
        heap.pop();
        int ver = k.second, dis=k.first;
        if(st[ver]) continue;
        st[ver] = true;
        for(int i=h[ver]; i!=-1; i=ne[i])
        {
            int j = e[i];
            if(dist[j] > dis + w[i])
            {
                dist[j] = dis + w[i];
                heap.push({dist[j], j});
            }
        }
    }
}

int main()
{
    memset(h, -1, sizeof h);
    cin >> n >> m;
    while(m--)
    {
        int x, y, c;
        cin >> x >> y >> c;
        add(x, y, c);
    }
    dij();
    for(int i=1; i<n; i++)
    {
        if(dist[i]!=inf)
        {
            cout << dist[i] << " ";
        }
    }
    return 0;
}
/*#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 5;
const int inf = 0x3f3f3f3f;
typedef pair<int, int> PII;
int h[N], e[N], ne[N], idx;
int w[N];
int dist[N];
bool st[N];
int n, m;
void add(int x, int y, int c)
{
    w[idx] = c, e[idx] = y, ne[idx] = h[x], h[x] = idx++;
}
void dijkstra()
{
    memset(dist, 0x3f, sizeof(dist));
    dist[0] = 0;
    for (int i = 0; i < n; i++)
    {
        int t = -1;
        for (int j = 0; j < n; j++)
        {
            if (!st[j] && (t == -1 || dist[j] < dist[t]))
                t = j;
        }
        st[t] = true;
        for (int j = h[t]; j != -1; j = ne[j])
        {
            int k = e[j];
            if (dist[k] > dist[t] + w[j])
            {
                dist[k] = dist[t] + w[j];
            }
        }
    }
}

int main()
{
    memset(h, -1, sizeof h);
    cin >> n >> m;
    while (m--)
    {
        int x, y, c;
        cin >> x >> y >> c;
        add(x, y, c);
    }
    dijkstra();
    for (int i = 1; i < n; i++)
    {
        if (dist[i] != inf)
        {
            cout << dist[i] << " ";
        }
    }
    return 0;
}*/