最短路径算法——无权最短路径——python3实现

本文参考来自数据结构与算法分析 java语言描述。

问题描述

现有一个有向无权图。如下图所示：

问题：使用某个顶点s作为输入参数，找出从s到所有其他顶点的最短路径。
说明：因为是无权图，因此我们可以为每台边赋值为1。这里选择v3为s作为起点。

问题分析

此时立刻可以说，从s到v3的最短路径是长为0的路径，标记此信息，得到下图。

现在开始寻找从s出发距离为1的顶点。这些顶点肯定是与s邻接的顶点，很明显，v1，v6从s出发只需要一条边就到了。所以，从s出发距离为1的顶点，为v1,v6。

现在开始寻找从s出发距离为2的顶点。这些顶点肯定是与v1,v6（距离为1的顶点）邻接的顶点。发现与v1邻接的顶点为v2,v4，与v6邻接的顶点没有（不能往回走，没有出边）。所以，从s出发距离为2的顶点，为v2,v4。

最后，考察与v2,v4邻接的顶点，即v5,v7。所以，从s出发距离为3的顶点，为v5,v7。

这种搜索图的方法称为广度优先搜索（breadth-first search）。按层处理顶点，距离起点近的顶点先处理，距离起点远的后处理。

伪代码（处理节点）

void unweighted(Vertex s){
    Queue<Vertex> q = new Queue<Vertex>();
    //把每个顶点的距离设为无穷大
    for each Vertex v
        v.dist = INFINITY
    //将起点的距离设为0
    s.dist = 0;
    //起点入队，作为算法的开始
    q.enqueue(s);
    //只要队列不为空，便继续循环
    while( !q.isEmpty() ){
        //获得出队顶点
        Vertex v = q.dequeue();
        //对与v邻接的每个顶点进行处理
        for each Vertex w adjacent to v
            if(w.dist == INFINITY){
                w.dist = v.dist + 1;
                w.path = v;//代表w的上一个经过的顶点为v
                //完成操作后，便入队，以用来接着分析与w邻接的顶点们
                q.enqueue( w );
            }
    }
}

实现过程

从s开始到顶点的距离放到dv列里，pv列用来代表，当前行代表的顶点的上一个经过的顶点。known列代表此顶点已经被处理过了。
初始化时，将起点的距离设置为0，且所有的顶点都不是know的。

结合伪代码进行分析：
【1】当第一次循环中，出队的是v3（每次循环只出队一个顶点）
【2】而第一次循环结束时，就是上表中“v3出队后”的数据情况，如下
【3】此时，对v3的邻接的顶点们都作了处理，所以v3就从F变成了T（即已知）
【4】与v3邻接的顶点v1,v6都作了处理，dv都变成了1，pv都为v3
【5】而因为与v1,v6的邻接顶点都还没有开始处理呢，所以v1,v6的F还不能变成T

得到无权最短路径

通过观察图，可以发现有两条路径长为3的最短路径。
【1】v3 => v1 => v2 => v5
【2】v3 => v1 => v4 => v7
我们可以通过数据变化表的最终情况来找到这两条路径。

注意，第一行代表v1，以此类推。
以找到v3 => v1 => v2 => v5路径为例，过程如下：
【1】找到距离为0的顶点，0在且只在第三行，所以第一个顶点为v3
【2】找到距离为1且pv为v3的顶点，有第一行和第六行，这里必须选一个，这里选第一行，所以第二个顶点为v1
【3】找到距离为2且pv为v1的顶点，有第二行和第四行，这里选第二行，所以第三个顶点为v2
【4】找到距离为3且pv为v2的顶点，只有第五行，所以第四个顶点为v5
【5】找到距离为4且pv为v5的顶点，没有，结束。
其实，以上步骤，是给出了，在对顶点进行数据处理后，找出无权最短路径的算法的思想。
其实可以，维护一些顶点间指针，用来指向下一个顶点，这样就可以用递归的思路来做，从起点开始，每递归到下一层距离dv便加1，用一个中间变量存储经过的顶点，每调用一次递归，便打印这个中间变量，这样，便能得到所有的无权最短路径。
这里得到无权最短路径的伪代码也不给出了，以上分析供大家理解参考。

代码实现

纸上得来终觉浅，绝知此事要躬行！还是觉得用代码实现一遍比较好。

from queue import Queue
class Vertex:
    #顶点类
    def __init__(self,vid,outList):
        self.vid = vid#出边
        self.outList = outList#出边指向的顶点id的列表，也可以理解为邻接表
        self.know = False#默认为假
        self.dist = float('inf')#s到该点的距离,默认为无穷大
        self.prev = 0#上一个顶点的id，默认为0

#创建顶点对象
v1=Vertex(1,[2,4])
v2=Vertex(2,[4,5])
v3=Vertex(3,[1,6])
v4=Vertex(4,[3,5,6,7])
v5=Vertex(5,[7])
v6=Vertex(6,[])
v7=Vertex(7,[6])
#创建一个长度为8的数组，来存储顶点，0索引元素不存
vlist = [False,v1,v2,v3,v4,v5,v6,v7]
def unweighted():
    #起点为v3
    vlist[3].dist = 0
    q = Queue()
    q.put(vlist[3])
    while(not q.empty()):
        v = q.get()#返回并删除队列头部元素
        for w in v.outList:
            if(vlist[w].dist == float('inf')):
                vlist[w].dist = v.dist + 1
                vlist[w].prev = v.vid
                q.put(vlist[w])

unweighted()
print('v1.prev:',v1.prev,'v1.dist',v1.dist)
print('v2.prev:',v2.prev,'v2.dist',v2.dist)
print('v3.prev:',v3.prev,'v3.dist',v3.dist)
print('v4.prev:',v4.prev,'v4.dist',v4.dist)
print('v5.prev:',v5.prev,'v5.dist',v5.dist)
print('v6.prev:',v6.prev,'v6.dist',v6.dist)
print('v7.prev:',v7.prev,'v7.dist',v7.dist)

运行结果：

与数据变化表的最终情况一致。
这里你可能会问，Vertex类的init函数中，明明有know成员，为什么在程序没有使用know成员（在处理节点后，就把该节点的know置为Ture），因为if(vlist[w].dist == float('inf'))的判断就相当于判断节点的know是否为Ture，因为一个已知的节点，它的距离就肯定不是无穷大了。
然后再使用递归，打印出所有可能的最短路径，把以下代码和以上代码合在一起就可以了。

traj_list = [3]#v3是起点直接加上

def print_traj(dist):
    last = traj_list[-1]
    print(traj_list,'该路径的长度为：',vlist[last].dist)
    temp_list = []#存储下一步的选项
    for i in range(1,len(vlist)):
        v = vlist[i]
        if((v.dist==dist) and (v.prev==last)):
            temp_list.append(i)
    if(len(temp_list)==0):
        return#终点
    #递归每个选项
    for i in temp_list:#i为顶点的索引
        traj_list.append(i)
        print_traj(dist+1)
        traj_list.pop()


print_traj(1)