堆优化的Dijkstra算法

  堆优化的Dijkstra算法

  迪杰斯特拉算法(Dijkstra)是由荷兰计算机科学家狄克斯特拉于1959 年提出的,因此又叫狄克斯特拉算法。是从一个顶点到其余各顶点的最短路径算法,解决的是有权图中最短路径问题。迪杰斯特拉算法主要特点是从起始点开始,采用贪心算法策略,每次遍历到始点距离最近且未访问过的顶点的邻接节点,直到扩展到终点为止。

  这里的Dijkstra算法是经过堆优化的算法,用于解决稀疏图问题。

  基本思想是通过邻接表储存每个点所能到达的点。用一个dist[]数组来储存初始点到达每个节点的最短距离,并初始化为无穷大。首先将初始点推入队列中,然后通过这个点来遍历每一个子节点并将每个使得初始节点到当前点的距离加上这个点到每个子节点的距离小于当前dist[]的值替换掉,并推入队列中,由于这里是小根堆,所取到的队首的值一定是当初始节点距离的点并开始下一轮遍历,以此类推直至找到目标节点的最短距离。

代码:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef pair<int,int> PII;
const int N=200010;
int ne[N],h[N],e[N],idx;//邻接表储存方式 
int n,m;
int w[N],dist[N];//w储存每条边的权值,dist储存初始节点到每一个节点的最短距离 
bool st[N];//储存每个点是否找到了最短距离 
priority_queue<PII,vector<PII>,greater<PII>> heap;//小根堆的定义 
void insert(int a,int b,int c)
{
    e[idx]=b;
    w[idx]=c;//储存权值 
    ne[idx]=h[a];
    h[a]=idx++;
}
int dijkstra()
{
    memset(dist,0x3f,sizeof dist);//将距离初始初始化为正无穷 
    heap.push({0,1});//将初始节点推入队列中 
    while(!heap.empty())
    {
        auto t=heap.top();
        heap.pop();
        int cnt=t.second,distance=t.first;//t.first代表当前节点到初始节点的最短距离,second代表当前是哪一个节点 
        if(st[cnt]) continue; 
        st[cnt]=true;
        for(int i=h[cnt];i!=-1;i=ne[i])//遍历这个链表 
        {
            int j=e[i];
            if(dist[j]>distance+w[i])//如果当前节点到子节点的距离加上这条边的权值小于之前头结点到子节点的距离 
            {
                dist[j]=distance+w[i];
                heap.push({dist[j],j});
            }
        }
    }
    if(dist[n]!=0x3f3f3f3f) return dist[n]; 
    return -1;
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        insert(a,b,c);
    }
    printf("%d",dijkstra());
    return 0;
}
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值