dijskra算法

观前提示:请不要在负权图中熟练的运用这个算法。

Dijskra的主要思路:

d数组记录源点到有向图上任意一点距离,其中源点到自身距离为0,到其他点距离为INF。将源点入队,并重复以下步骤:

  1. 把源点标记为走过,找到源点能到达的所有点中距他最近的那个点。

  2. 同时取那个点本身的数和新路的较小值。

所以,和 BFS很像。

1

首先给出没有优化的Dijskra,思路比较清晰,只是复杂度挺高的。

#include<bits/stdc++.h>
using namespace std;
#define INF (1<<31)-1
struct Edge{
    int to,next,value;
}edge[1000005];
bool q[20005];
int tot,head[20005],len[20005];
inline void add(int u,int v,int w){
    tot++,edge[tot].to=v,edge[tot].value=w,edge[tot].next=head[u],head[u]=tot;
}
int main(){
    int n,m,fuck,u,v,w,now,minn=INF;
    read(n),read(m),read(fuck);
    for(register int i=1;i<=n;i++)
    	len[i]=INF;
    for(register int i=0;i<m;i++)
        read(u),read(v),read(w),add(u,v,w);
    len[fuck]=0,now=fuck;
    while(!q[now]){
    	q[now]=1;
        for(register int i=head[now];i;i=edge[i].next)
            if(!q[edge[i].to]&&len[edge[i].to]>len[now]+edge[i].value)
                len[edge[i].to]=len[now]+edge[i].value;
        minn=INF;
        for(register int i=1;i<=n;i++)
            if(!q[i]&&minn>len[i])
                minn=len[i],now=i;
    }
    for(register int i=1;i<=n;i++)
        cout<<len[i]<<" ";
}

2

加了堆优化就和BFS用数组实现改成用STL的队列实现很像。

使用了stl优先队列 :priority_queue<pair<int, int> > q;

代码

inline void dijskra(){
    d[start]=0;
    q.push(make_pair(0,start));
    while(!q.empty())
    {
        int x=q.top().second;  q.pop();
        if(yes[x]) continue;
        yes[x]=1;
        for(register int i=head[x];i;i=edge[i].next){
            int y=edge[i].to;
            int z=edge[i].value;
            if(d[y]>d[x]+z){
                d[y]=d[x]+z;
                q.push(make_pair(-d[y],y));
            }
        }
    }
}
posted @ 2020-03-19 18:50  iloveori  阅读(4704)  评论(1编辑  收藏  举报