复习图论基础——dijkstra朴素写法

 一、情景带入

首先我们引入下面的问题，这只是一种具体的应用，如果学会题目的用法一般的问题也就都能解决了。

给定一个 n 个点 m 条边的有向图，图中可能存在重边和自环，所有边权均为正值。

请你求出 1 号点到 n 号点的最短距离，如果无法从 1 号点走到 n 号点，则输出 −1。

输入格式

第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z，表示存在一条从点 x 到点 y 的有向边，边长为 z。

输出格式

输出一个整数，表示 1 号点到 n 号点的最短距离。

如果路径不存在，则输出 −1。

数据范围

1≤n≤5001≤n≤500,
1≤m≤1051≤m≤105,
图中涉及边长均不超过10000。

输入样例：

3 3
1 2 2
2 3 1
1 3 4

输出样例：

3

 

 二、算法思路：

初始时，S只包含源点，即S＝{v}，dis[v]为0。U包含除v外的其他顶点，即:U={其余顶点}=N-S，若v与U中顶点u有边，则<u,v>正常有权值，若u不是v的出边邻接点，则<u,v>权值为∞。从U中选取一个距离v最小的顶点k，把k加入S中。用k为中间的跳板更新dis[N]，且k加入S就代表dis[k]已经是最小值了，再重复上述过程直到所有点都在S中。

举一个简单的例子：

如果是上图中的情况，那么第一次加到S中的点（假设源点1已经加入到S中）就是距离S中的点距离最近的点5。然后用5去更新dis[N]。

接着上述步骤，找到N-S中距离S中的点的距离最小的点，很明显应该将点4加入S。

一直重复上述步骤，直到所有点都加入到S中。

tips:上面研究的都是有向图，其实无向图也是一样的。可以将无向图理解为特殊的有向图（a->b,同时b->a），也就是e[i][j]赋值时同时赋值e[j][i]就i能满足条件。

三、代码：

#include<iostream>
#include<cstring>
using namespace std;

const int N = 510;
int dis[N],mark[N];//用于标记该点是否走过0没1走。
int e[N][N];
int n,m;

void dijkstra()
{
    memset(dis,0x3f,sizeof dis);
    dis[1]=0;
    for(int i=1;i<=n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(mark[j]==0&&(t==-1||dis[j]<dis[t]))
                t=j;
        mark[t]=1;        
        for(int k=1;k<=n;k++)
            dis[k]=min(dis[k],dis[t]+e[t][k]);
    }
}

int main(void)
{
    memset(e,0x3f,sizeof e);
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        e[a][b]=min(e[a][b],c);
    }
    dijkstra();
    if(dis[n]!=0x3f3f3f3f)
    printf("%d\n",dis[n]);
    else
    printf("-1\n");
}

如果单单为了解决我所举例的这道题目，那么还可以在dijkstra函数底部增加一个判断条件if(t==n) break;减少循环次数。

四、总结

其实该算法还可以用堆来进行优化，时间复杂度会进一步降低，我会找个时间将优化的思路和代码整理好写入该篇。