Dijkstra

· · 算法·理论

原题:P4779 【模板】单源最短路径(标准版)
#include<bits/stdc++.h>
#define MAXN 100005
#define MAXM 200005
#define INF 2147483647
using namespace std;
struct Edge
{
    int u,v,w,nxt;
}e[MAXM];
int head[MAXN],cnt,n,m,start,vis[MAXN],dis[MAXN];
struct node
{
    int w,now;
    inline bool operator <(const node &x)const
    {
        return x.w<w;
    }
};
priority_queue<node>q;
inline void add(int u,int v,int w)
{
    e[++cnt].v=v;
    e[cnt].w=w;
    e[cnt].nxt=head[u];
    head[u]=cnt;
}
void dijkstra()
{
    for(int i=1;i<=n;++i) dis[i]=INF;
    dis[start]=0;
    q.push(node{0,start});
    while(!q.empty())
    {
        node t=q.top();
        q.pop();
        int u=t.now;
        if(vis[u]) continue;
        vis[u]=1;
        for(int i=head[u];i;i=e[i].nxt)
        if(dis[e[i].v]>dis[u]+e[i].w)
        {
            dis[e[i].v]=dis[u]+e[i].w;
            q.push((node){dis[e[i].v],e[i].v});
        }
    }
}
int main()
{
    cin>>n>>m>>start;
    for(int i=1;i<=m;++i)
    {
        int u,v,w;
        cin>>u>>v>>w;
        add(u,v,w);
    }
    dijkstra();
    for(int i=1;i<=n;++i) printf("%d ",dis[i]);
    return 0;
}