P3371 【模板】单源最短路径

题目描述

如题，给出一个有向图，请输出从某一点出发到所有点的最短路径长度。

输入输出格式

输入格式：

第一行包含三个整数N、M、S，分别表示点的个数、有向边的个数、出发点的编号。

接下来M行每行包含三个整数Fi、Gi、Wi，分别表示第i条有向边的出发点、目标点和长度。

输出格式：

一行，包含N个用空格分隔的整数，其中第i个整数表示从点S出发到点i的最短路径长度（若S=i则最短路径长度为0，若从点S无法到达点i，则最短路径长度为2147483647）

输入输出样例

输入样例#1：  复制

4 6 1
1 2 2
2 3 2
2 4 1
1 3 5
3 4 3
1 4 4

输出样例#1：  复制

0 2 4 3

说明

时空限制：1000ms,128M

数据规模：

对于20%的数据：N<=5，M<=15

对于40%的数据：N<=100，M<=10000

对于70%的数据：N<=1000，M<=100000

对于100%的数据：N<=10000，M<=500000

样例说明：

图片1到3和1到4的文字位置调换

#include<bits/stdc++.h>

using namespace std;

const int MAXN = 500000 + 10;
int First[MAXN] , Next[MAXN] ;
int dis[10010] , book[10010], n , m , k, inf;

typedef pair<int , int > pii;
priority_queue<pii , vector<pii> , greater<pii> >q;

struct f
{
    int u;
    int v;
    int w;
}a[MAXN];

void dijkstra(int x)
{
    dis[x] = 0;
    q.push(make_pair(0,x));
    int num = 0;
    while(!q.empty() && num < n)
    {
        int d = q.top().first , u = q.top().second;
        //cout << u << endl;
        q.pop();
        if(book[u])
            continue;
        book[u] = 1;
        num ++;
        for(int k = First[u] ; k != -1; k = Next[k])
        {
            if(dis[ a[k].v ] > a[k]. w + dis[u])
            {
                dis[ a[k].v ] = a[k].w + dis[u];
                q.push(make_pair(dis[ a[k].v ] , a[k].v));
            }
        }
    }
}
int main()
{
    inf = 0x7fffffff;
    cin >> n >> m >> k;
    memset(First, -1, sizeof(First));
    for(int i = 1; i <= m ; i ++)
    {
        scanf("%d%d%d" , &a[i].u, &a[i].v, &a[i].w);
        Next[i] = First[ a[i].u ] ; First[ a[i].u ] = i;
    }
    for(int i = 1; i <= n ; i ++)
        dis[i] = inf;
    dijkstra(k);
    cout << dis[1];
    for(int i = 2; i <= n ; i ++)
        cout << " " << dis[i];
    return 0;
}