python解决图的最短路径问题
遇到一个算法题目,描述如下:
对图结构有了解的不难发现,这是经典的求图的最短路径问题。以下是python代码:
def findMin(row): minL = max(row) for i in row: if i != -1 and minL > i: minL = i return minL def initRow(row, plus): r = [] for i in row: if i != -1: i += plus r.append(i) return r def getMinLen(table, e, t): count = len(table) - 1 startPoint = 1 #记录原点到各点最短距离 初始值为-1,即不可达 lenRecord = list((-1 for i in range(count+1))) lenRecord[startPoint] = 0 #记录每次循环的起点 points = [startPoint] #已得到最短距离的点 visited = set() while len(points)>0: #当前起点 curPoint = points.pop() #原点到当前起点的距离 curLen = lenRecord[curPoint] #当前起点到各点的距离 curList = initRow(table[curPoint], t) #当前起点到各点的最短距离 curMin = findMin(curList) visited.add(curPoint) idx = 0 while idx<count: idx += 1 #当前点不可达或到当前点的最短距离已计算出 则跳过 if curList[idx] == -1 or idx in visited: continue #记录距离当前起点最近的点作为下次外层循环的起点 if curList[idx] == curMin: points.append(idx) #如果从原点经当前起点curPoint到目标点idx的距离更短,则更新 if lenRecord[idx] == -1 or lenRecord[idx] > (curLen+curList[idx]): lenRecord[idx] = curLen+curList[idx] return lenRecord[e] def processInput(): pointCnt, roadCnt, jobCnt = (int(x) for x in raw_input().split()) table = [] for i in range(pointCnt+1): table.append([-1] * (pointCnt+1)) for i in range(roadCnt): (x, y, w) = (int(n) for n in raw_input().split()) if table[x][y] == -1 or table[x][y] > w: table[x][y] = w table[y][x] = w res = [] for i in range(jobCnt): e, t = (int(x) for x in raw_input().split()) res.append(getMinLen(table, e, t)) for i in res: print(i) processInput()