y总的堆优化版dijkstra算法模板

最新推荐文章于 2024-09-14 22:25:23 发布

天才少年伊泽瑞尔

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量192

点赞数 1

分类专栏： ACM steps 文章标签：算法图论数据结构 1024程序员节

本文链接：https://blog.csdn.net/zhao_xin_ran/article/details/127876724

版权

ACM steps 专栏收录该内容

27 篇文章 0 订阅

订阅专栏

优先队列+链式前向星存图，用于稀疏图的单源最短路径，时间复杂度为O（mlogn)。

模板题见洛谷【模板】单源最短路径（标准版） - 洛谷

弱化标准版都能AC

#include<bits/stdc++.h>

using namespace std;
#define INF 2147483647

typedef pair<int,int> PII;

const int N=1e5+5;
int head[N];
struct node
{
    int to,dis,next;
}edge[200005];
int dist[N];
bool st[N];
int n,m,s,cnt;

void add(int from,int to,int dis)
{
    edge[cnt].to=to;
    edge[cnt].dis=dis;
    edge[cnt].next=head[from];
    head[from]=cnt++;
}

void dijkstra()
{
    for(int i=1;i<=n;i++)
        dist[i]=INF;
    dist[s]=0;
    
    priority_queue<PII ,vector<PII> ,greater<PII> > t;
    t.push({0,s});
    while(t.size())
    {
        PII temp=t.top();
        t.pop();

        int distance=temp.first,vec=temp.second;
        if(st[vec])
            continue;
        st[vec]=true;

        for(int i=head[vec];i != -1;i=edge[i].next)
        {
            int j=edge[i].to;
            if(dist[j] > dist[vec]+edge[i].dis)
            {
                dist[j]=dist[vec]+edge[i].dis;
                t.push({dist[j],j});
            }
        }
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&s);
    memset(head,-1,sizeof head);
    while(m--)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
    }
    dijkstra();
    for(int i=1;i<=n;i++)
    {
        cout<<dist[i]<<' ';
    }
}