利用Dijkstra算法实现记录每个结点的所有最短路径

Posted 我们都曾拥有最美的时光

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了利用Dijkstra算法实现记录每个结点的所有最短路径相关的知识,希望对你有一定的参考价值。

最近在做PAT时发现图论的一些题目需要对多条最短路径进行筛选,一个直接的解决办法是在发现最短路径的时候就进行判断,选出是否更换路径;另一个通用的方法是先把所有的最短路径记录下来,然后逐个判断。前者具有一定的难度并且不好排查BUG,因此我设计了一种基于Dijkstra的记录所有最短路的简捷算法,用于解决此类题目。

我们知道,Dijkstra是解决单源最短路问题的,并且最基本的算法仅能求出最短路的长度,而不能输出路径,本文基于Dinjkstra进行改进,使之能记录源点到任意点的所有最短路径。

使用vector<int>来记录一条路径,因为每个结点可能有多条最短路径,因此把这些路径都装在一个vector中,因此可以用一个vector<vector<int> >来表示一个结点的所有最短路径,把所有结点的最短路径都存放起来,又需要一个vector容器,因此所有结点的所有最短路径的集合可以用vector<vector<vector<int> > >来表示。

约定:结点编号为0到N-1,源点为0,到每个点的最短距离存储在数组minD[N]中。

在Dijkstra算法初始化时,找出所有源点的邻接点w并且把相应的最短距离minD[w]更新,同时初始化这些点w的第一条最短路径0->w(实现方法为分别push_back 0和w)。接下来将会找到一个到源点最短的点v,并且把v并入集合,对v的所有未访问的邻接点,如果到达w的路径(0->...->w)在包含v之后(0->...->v->w)变短,则删除w之前所有的最短路径,并且更新为到v的所有最短路径加上w点(注意对每个到v的最短路径都要这样处理);如果到达w的路径在包含v之后长度不变,说明发现了一条新的最短路径,在w原来最短路径容器的基础上再压入一个新的最短路径,这条路径为所有到v的最短路径加上w点。

经过这样的运算,就可以得到所有结点的所有最短路径了,下面以一个实例对算法进行测试,并且附上源代码。

题目:求下图的源点0到所有结点的最短路径。


输入:

5 8
2 4 1
0 1 3
0 2 6
1 3 2
1 4 1
3 4 1
3 2 1
0 4 4

输出:



源码为:

#include <iostream>
#include <stdio.h>
#include <vector>

using namespace std;

#define MAX 1001
#define INF 99999999

int G[MAX][MAX];
int minD[MAX];
int minDist;
int finalSet[MAX];

int main()

    int N,M;
    int v1,v2;
    int len;
    cin >> N >> M;
    for(int i = 0; i < N; i++)
        finalSet[i] = 0;
        minD[i] = INF;
        for(int j = 0; j < N; j++)
            G[i][j] = INF;
    
    for(int i = 0; i < M; i++)
        scanf("%d%d%d",&v1,&v2,&len);
        G[v1][v2] = G[v2][v1] = len;
    

    vector<vector<vector<int> > > nodes(N);

    // 设0为源点,计算从0到所有点的所有最短路径
    finalSet[0] = 1;
    minD[0] = 0;
    // 首先把所有源点邻接点的最短距离初始化为源点到这些点的距离
    for(int i = 1; i < N; i++)
        if(G[0][i] != INF) 
            minD[i] = G[0][i];
            vector<vector<int> > minPaths;
            minPaths.clear();
            vector<int> pathList;
            pathList.clear();
            pathList.push_back(0);
            pathList.push_back(i);
            minPaths.push_back(pathList);
            nodes[i] = minPaths;
        
    

    // 从所有minD中找出最小的,并入集合S,重复N-1次(源点已经加入集合)
    for(int i = 1; i < N; i++)
        minDist = INF;
        int v = -1; // 记录到源点记录最小的结点
        for(int w = 1; w < N; w++)
            if(!finalSet[w] && minDist > minD[w])
                minDist = minD[w];
                v = w;
            
        
        if(v == -1) break; // v = -1说明找不到点了,当图不连通时才会出现这种情况
        // 已经找到了到源点最近的点v,将其并入集合,并且考虑原来的最短距离0->...->W在加入了v之后有没有可能变短
        // 如果变短了,就更新为0->...>v->W
        finalSet[v] = 1;
        for(int w = 1; w < N; w++)
            if(!finalSet[w])
                int newD = minDist + G[v][w];
                if(newD < minD[w])
                    minD[w] = newD;
                    vector<vector<int> > minPathsV = nodes[v];
                    vector<int> pathList;
                    nodes[w].clear();
                    for(int index = 0; index < minPathsV.size(); index++)
                        pathList = minPathsV[index];
                        pathList.push_back(w);
                        nodes[w].push_back(pathList);
                    

                else if(newD == minD[w])

                    vector<vector<int> > minPathsV = nodes[v];
                    vector<int> pathList;
                    for(int index = 0; index < minPathsV.size(); index++)
                        pathList = minPathsV[index];
                        pathList.push_back(w);
                        nodes[w].push_back(pathList);
                    

                
            
        
    
    for(int i = 1; i < N; i++)
        cout << "------------" << endl;
        cout << "0 to "<< i << ":" << endl;
        cout << "The miniest distance:" << endl << minD[i] << endl;
        cout << "The possible paths:" << endl;
        vector<vector<int> >minPaths = nodes[i];
        int size = minPaths.size();
        vector<int> pathList;
        for(int j = 0; j < size; j++)
            pathList = minPaths[j];
            int pathSize = pathList.size();
            for(int k = 0; k < pathSize - 1; k++)
                cout << pathList[k] << "->";
            
            cout << pathList[pathSize - 1] << endl;
        

    

    return 0;




以上是关于利用Dijkstra算法实现记录每个结点的所有最短路径的主要内容,如果未能解决你的问题,请参考以下文章

dijkstra算法怎么记录路径

简单的图论问题之单源最短路dijkstra算法

Dijkstra算法思想

关于Dijkstra算法

Dijkstra算法

图论 最短路径算法 2.Dijkstra算法