python --Dijkstra 算法求取最短路径

#大名鼎鼎的Dijkstra是一种广度优先算法：

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import numbers
import numpy
import math

'''
Dijkstra 算法说明
初始条件：
1、设定邻接列表或邻接矩阵
2、选定初始节点
3、设立已确立最短线路的节点集，包括目标节点标志、路径权重、路径节点序列（set1）
4、设立待优化路径集合，包裹目标节点标志、路径权重、路径节点序列（set2）
5、当前节点
*********************************
方法：
1、选定节点该，直接进入set1
2、当前节点不包含于set1的邻接节点，并累计到这些这些节点的权重的到临时settemp；
3、settemp中的元素逐个与set2中的待优化路径比较：目标节点相同则相比较区中，set中的做选择权重小的路径；未有匹配的路径则直接写入（松弛）
4、在set2中选择权重最小的节点写入set1，并将这个节点作为当前节点
5、重复2-4部

#代码：
def dijkstra(curp,setf,nfix):
        setf[curp[-1][0]]=[curp[-1][1],curp[-1][2]]
        temppoint = ''
        temppath = ''
        tempright = 0
        for i in adjoin:
            if i[0] ==curp[0][0] and i[1] not in setf:
                temppoint=i[1]
                temppath=curp[0][1]+i[1]
                tempright=i[2]+curp[0][2]
            else:
                if i[1]==curp[0][0] and i[0] not in setf:
                    temppoint = i[0]
                    temppath = curp[0][1] + i[0]
                    tempright = i[2] + curp[0][2]
                else:temppoint=None

            if temppoint!=None:
                if temppoint not in nfix:
                    nfix[temppoint]=[temppath,tempright]
                else:
                    if nfix[temppoint][1]>tempright:
                        nfix[temppoint]=[temppath,tempright]

        curp.clear()
        curp.append(['','',1000])
        for key in nfix:
            if nfix[key][1]<curp[-1][2]:
                curp.clear()
                curp.append([key,nfix[key][0],nfix[key][1]])

        setf[curp[-1][0]]=[curp[-1][1],curp[-1][2]]
        del nfix[curp[-1][0]]
        if nfix :
            dijkstra(curp,setf,nfix)
        else:
            print(setf)

#用例
adjoin=[]
adjoin.append(['A','B',10])
adjoin.append(['A','F',11])
adjoin.append(['B','G',16])
adjoin.append(['B','C',18])
adjoin.append(['B','I',12])
adjoin.append(['C','I',8])
adjoin.append(['C','D',22])
adjoin.append(['D','I',21])
adjoin.append(['D','H',16])
adjoin.append(['D','E',20])
adjoin.append(['D','G',24])
adjoin.append(['E','F',26])
adjoin.append(['F','G',17])
adjoin.append(['G','H',19])
startpoint='A'
currentpoint=[['B','B',0]]
setfixed={}#setfixed{}={'B':['AB',10]}
needfix={}#与setfixed[0]相同

dijkstra(currentpoint,setfixed,needfix)
#输出：
{'B': ['B', 0], 'A': ['BA', 10], 'I': ['BI', 12], 'G': ['BG', 16], 'C': ['BC', 18], 'F': ['BAF', 21], 'D': ['BID', 33], 'H': ['BGH', 35], 'E': ['BAFE', 47]}