Dijkstra模板

Posted 2021-02-17 hdgs3-blog

模板题

#include<bits/stdc++.h>

using namespace std;

int n,m,s;
int head[100010],size,dis[100010];
bool falg[100010];

struct edge
{
    int to,next,val;
}e[200010];

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

void add_edge(int,int,int);

void Dij();

int main()
{
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<=m;i++)
    {
        int a,b,v;
        scanf("%d%d%d",&a,&b,&v);
        add_edge(a,b,v);
    }
    Dij();
    for(int i=1;i<=n;i++)printf("%d ",dis[i]);
return 0;
}

void add_edge(int a,int b,int val)
{
    e[++size].to=b;
    e[size].val=val;
    e[size].next=head[a];
    head[a]=size;
}

void Dij()
{
    memset(dis,0x3f,sizeof(dis));
    dis[s]=0;
    q.push(make_pair(dis[s],s));
    while(!q.empty())
    {
        int u=q.top().second;
        q.pop();
        if(falg[u]==true)continue;
        falg[u]=true;
        for(int i=head[u];i!=0;i=e[i].next)
        {
            int to=e[i].to;
            int val=e[i].val;
            if(dis[to]>dis[u]+val)
            {
                dis[to]=dis[u]+val;
                q.push(make_pair(dis[to],to));
            }
        }
    }
}