堆优化的Dijkstra算法

 1.

  vector

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define inf 0x3f3f3f3f
#define maxn 1024
using namespace std;

struct node
{
    int to,dis;
    node (int a,int b){to=a,dis=b;}
    friend bool operator <(node a,node b)
    {
        if(a.dis==b.dis) return a.to<b.to;
        return a.dis>b.dis;
    }
};

vector<node> mp[maxn];
int dis[maxn],vis[maxn];
int n,m,st,ed;

void Dijkstra()
{
    memset(dis,inf,sizeof(dis));
    dis[st]=0;
    priority_queue<node> que;
    que.push(node(st,dis[st]));
    while(!que.empty())
    {
        node x=que.top();
        que.pop();
        for(int i=0;i<mp[x.to].size();i++)
        {
            node y=mp[x.to][i];
            if(dis[y.to]>x.dis+y.dis)
            {
                dis[y.to]=x.dis+y.dis;
                que.push(node(y.to,dis[y.to]));
            }
        }
    }
    if(dis[ed]<inf)
        printf("%d\n",dis[ed]);
    else printf("-1\n");
}

int  main()
{
    int a,b,d;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=0;i<=n;i++)
            mp[i].clear();
        while(m--)
        {
            scanf("%d%d%d",&a,&b,&d);
            mp[a].push_back(node(b,d));
            mp[b].push_back(node(a,d));//反路不一定通,可能是有向图
        }
        scanf("%d%d",&st,&ed);
        Dijkstra();
    }
    return 0;
}

2.

链式前向星

#include<bits/stdc++.h>
#define maxn 10007
#define inf 0x3f3f3f
using namespace std;

struct node
{
    int to,dis;
    node (int a,int b){to=a,dis=b;}
    friend bool operator <(node a,node b)
    {
        if(a.dis==b.dis) return a.to<b.to;
        return a.dis>b.dis;
    }
};

struct edge
{
    int to,dis,next;
}mp[maxn];

int dis[maxn],vis[maxn],head[maxn];
int n,m,st,ed,ecnt;

void add(int a,int b,int d)
{
    mp[ecnt]={b,d,head[a]};
    head[a]=ecnt++;
}

void Dijkstra()
{
    memset(dis,inf,sizeof(dis));
    dis[st]=0;
    priority_queue<node> que;
    que.push(node(st,dis[st]));
    while(!que.empty())
    {
        node x=que.top();
        que.pop();
        for(int i=head[x.to];i!=-1;i=mp[i].next)
        {
            edge y=mp[i];
            if(dis[y.to]>x.dis+y.dis)
            {
                dis[y.to]=x.dis+y.dis;
                que.push(node(y.to,dis[y.to]));
            }
        }
    }
    if(dis[ed]<inf)
        printf("%d\n",dis[ed]);
    else printf("-1\n");
}

int  main()
{
    int a,b,d;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(head,-1,sizeof(head));
        ecnt=0;
        while(m--)
        {
            scanf("%d%d%d",&a,&b,&d);
            add(a,b,d);
            add(b,a,d);//反向不一定通,可能是有向图
        }
        scanf("%d%d",&st,&ed);
        Dijkstra();
    }
    return 0;
}

 

转载于:https://www.cnblogs.com/wangtao971115/p/10358326.html