python 狄克斯特拉算法

前言

狄克斯特拉算法是解决加权图求最短路径的算法,广度优先算法可以求非加权图的最短路径,但是如果图的边权重不一样,那么就可以用狄克斯特拉算法来解决。

背景

现有一问题,想要求A到D之间的最短距离,其中每条边的数字代表距离,如下图:



很显然,如果使用广度优先算法是求不出正确结果的。
狄克斯特拉算法描述:

  1. 找出距离最短的节点,
  2. 对于该节点的邻居,如果有前往他们更短距离,更新其开销,
  3. 重复1,2直到所有节点都更新了。

下面我们具体执行这些步骤:
首先建立一张表格,用于存放到达每个节点的开销



更新起点的时候由于还不知道到EFD的距离,因此我们用无穷大表示先。在计算的过程中,我们会不断地更新这张表。另外,为了找到最短路径,我们还需要添加有父节点的列


好,我们开始执行算法:

先找到最便宜的节点,根据我们建立的表格知道,是E,然后计算前往E的邻居的开销,也是就是前往F的。


下一个最便宜的节点是B,更新前往B的令居的开销,也就是更新前往D的开销


重复上述步骤
下一个最便宜的节点是F,更新前往F的令居的开销,也就是更新前往C,D的开销


下一个最便宜的节点是C,更新前往C的令居的开销,也就是更新前往D的开销



所有节点都更新完了,可以确定最短路径了,到D最短总开销是16,路径根据表可知,D的父节点C,C的父节点F,F的父节点E,E的父节点A,那么就是A-E-F-C-D。

算法实现

我们以下面简单的图为例,求从A到D的最短路径



要解决这个问题,需要维护三个散列表(python中dict):



后面会不断更新cost和parent,
首先实现graph:
graph = {}
graph['A'] = {"B":4,"C":1}
graph['B'] = {"D":3}
graph['C'] = {"B":2,"D":7}
graph['D'] = {}

实现cost:

costs = {}
costs['B'] = 4
costs['C'] = 1
costs['D'] = float('inf') # 表示无穷大

实现parent:

parents = {}
parents['B'] = 'A'
parents['C'] = 'A'
parents['D'] = None

最后需要一个list用于记录处理过的节点

processed = []

数据结构都建好了,那么我们按怎样的算法步骤执行呢,步骤:

  1. 获取离开始节点最近的节点
  2. 更新这个节点的邻居开销
  3. 如果有邻居被更新,那么同时更新其父节点
  4. 将该节点标记为处理过,并重复执行以上步骤,直到所有节点都已这样处理。

下面根据算法步骤列出代码:

graph = {}
graph['A'] = {"B":4,"C":1}
graph['B'] = {"D":3}
graph['C'] = {"B":2,"D":7}
graph['D'] = {}

costs = {}
costs['B'] = 4
costs['C'] = 1
costs['D'] = float('inf') # 表示无穷大

parents = {}
parents['B'] = 'A'
parents['C'] = 'A'
parents['D'] = None

processed = []

def get_min_cost_node(costs):
    min_cost = float('inf')
    min_cost_node = None
    for n in costs:
        cost = costs[n]
        if cost < min_cost and n not in processed:
            min_cost = cost
            min_cost_node = n
    return min_cost_node

node = get_min_cost_node(costs)  # 在未处理的节点中找到开销最小的节点
while node is not None:
    cost = costs[node]
    neighbs = graph[node]
    for n in neighbs:  # 遍历当前节点的所有邻居
        new_cost = cost + neighbs[n]
        if costs[n] > new_cost:  # 如果经由当前节点前往该邻居更近 就更新该邻居的开销
            costs[n] = new_cost
            parents[n] = node  # 同时更新该邻居节点的父节点
    processed.append(node)  # 将节点加入到已处理队列
    node = get_min_cost_node(costs)  # 找到下一步要处理的节点

print(processed)  #求出最短路径

附录

参考《算法图解》

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,802评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,109评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,683评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,458评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,452评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,505评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,901评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,550评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,763评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,556评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,629评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,330评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,898评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,897评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,140评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,807评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,339评论 2 342

推荐阅读更多精彩内容