洛谷 P3371 【模板】单源最短路径（弱化版） 题解 （dijkstra）

题目描述

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

输入格式

第一行包含三个整数 n,m,s，分别表示点的个数、有向边的个数、出发点的编号。

接下来 m 行每行包含三个整数 u,v,w，表示一条 u→v 的，长度为 w 的边。

输出格式

输出一行 n 个整数，第 i 个表示 s 到第 i 个点的最短路径，若不能到达则输出 2^{31}-1。

AC代码 (dijkstra)

#include<bits/stdc++.h>
using namespace std;

priority_queue< pair <int,int > >q;
int n,m,s,d[5000010];
bool vis[5000010];
int head[5000010],to[5000010],next[5000010],edge[5000010],cnt;

void add(int x,int y,int v)
{
    cnt++;
    to[cnt]=y;
    edge[cnt]=v;
    next[cnt]=head[x];
    head[x]=cnt;
}

void dijkstra()
{
    for (int i=1;i<=n;i++)
        d[i]=2147483647;
    memset(vis,0,sizeof(vis));
    d[s]=0;
    q.push(make_pair(0,s));
    while (q.size())
    {
        int x=q.top().second;
        q.pop();
        if (vis[x])
            continue;
        vis[x]=1;
        for (int i=head[x];i;i=next[i])
        {
            int y=to[i],v=edge[i];
            if (d[y]>d[x]+v)
            {
                d[y]=d[x]+v;
                q.push(make_pair(-d[y],y));
            }
        }
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&s);
    int x,y,v;
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&v);
        add(x,y,v);
    }
    
    dijkstra();
    
    for (int i=1;i<=n;i++)
        printf("%d ",d[i]);
    return 0;
}