013 图的实现

邻接矩阵(adjacency matrix)

有向图

假设有5个城市,名字依次叫做ABCDE。

vertex = ['A', 'B', 'C', 'D', 'E']  
  
adjacency_matrix = [  
    [0, 1, 0, 0, 0],  
    [1, 0, 0, 0, 1],  
    [0, 1, 0, 1, 0],  
    [0, 1, 1, 0, 0],  
    [0, 0, 0, 1, 0],  
]

adjacency_matrix[i][j] 表示从i到j是否具有有向边。

有向边的含义:

  • adjacency_matrix[2][1] 等于 1, 表明从C可以到达B。
  • adjacency_matrix[1][2] 等于 0, 表明从B不能到达C。
  • adjacency_matrix[4][3] 等于 1, 表明从E可以到达D。
  • adjacency_matrix[3][0] 等于 0, 表明从D不能到达A。

在这个例子中,每一个城市到达自己这里,好像说不过去,没有什么意义,索性让正对角线为0。

无向图

注意,如果这个图是无向的,那么邻接矩阵是对称矩阵,类似这样:

adjacency_matrix = [
    [0, 1, 0, 0, 1],
    [1, 0, 1, 1, 0],
    [0, 1, 0, 1, 1], 
    [0, 1, 1, 0, 0],
    [1, 0, 1, 0, 0]
]

带权(网)

如果ABCDE这5个城市之间修起了公路,想通过边的权重来代表两个城市之间的公路距离是多少千米。

带权的形式是这样的:

vertex = ['A', 'B', 'C', 'D', 'E']  

adjacency_matrix = [
    [0, 72, 0, 0, 23],
    [72, 0, 94, 61, 0],
    [0, 94, 0, 19, 59],
    [0, 61, 19, 0, 0],
    [23, 0, 59, 0, 0]  
]

在这里,A到B的公路距离是72km, B到C的公路距离是94km, B到D的公路距离是61km, A到E的公路距离是23km。

带多维权重信息(容器)

假设后来在公路的基础上,又修建了铁路,开辟了飞机航线。乘坐不同的交通工具会花不同的时间,假设单位为分钟数。

况且每个地方地势不一样,去的时候和回来的时候所花的时间也不一样。

可以通过一个类来表示多种形式的权。或者,元组或命名元组,dataclass类也可以。

这里使用dataclass可能更方便。

from dataclasses import dataclass

vertex = ['A', 'B', 'C', 'D', 'E']


@dataclass
class E:
    driving_time: int
    train_time: int
    flying_time: int


adjacency_matrix = [
    [E(0, 0, 0), E(4320, 1200, 60), E(0, 0, 0), E(0, 0, 0), E(1800, 600, 48)],
    [E(3600, 900, 54), E(0, 0, 0), E(4800, 1500, 60), E(3000, 1080, 36), E(0, 0, 0)],
    [E(0, 0, 0), E(5400, 1800, 66), E(0, 0, 0), E(2400, 720, 30), E(4200, 1320, 54)],
    [E(0, 0, 0), E(4200, 1500, 48), E(2100, 600, 24), E(0, 0, 0), E(0, 0, 0)],
    [E(1200, 480, 36), E(0, 0, 0), E(3900, 960, 42), E(0, 0, 0), E(0, 0, 0)]
]

E 数据类是Edge边的缩写,表示图中的一条边,存储了多种交通方式的时间权重,其中:

  • driving_time 表示开车时间
  • train_time 表示乘坐火车时间
  • flying_time 表示乘坐飞机时间

adjacency_matrix 中每个元素是一个 E 的实例,比如:

  • E(4320, 1200, 60) 表示从城市 A 到城市 B 的开车时间是4320分钟,乘火车1200分钟,乘飞机60分钟
  • E(0, 0, 0) 表示城市之间这一方向没有连接

邻接表(adjacency list)

有向图

前面有向图的例子,转换成邻接表是这样:

vertex = {  
  'A': ['B'],  
  'B': ['A', 'E'],  
  'C': ['B', 'D'],  
  'D': ['B', 'C'],  
  'E': ['D'],  
}

这里的ABCDE是字符串,实际上,可以是任意Python类型,但是用作字典键时,必须是可哈希的(hash)。

这里,'C': ['B', 'D'] 的意思是C可以到达B和D,也即存在这样一条有向边。

无向图

与邻接矩阵类似,如果图是无向图,A和B之间有边,那么需要保证A中有B,B中有A。就像这样:

vertex = {  
  'A': ['B'],  
  'B': ['A'],  
}

带权图(网)

带权的形式只需用一个元组来包裹即可。这里也可以使用命名元组等方式,都可以。

# 加权邻接表
vertex = {
  'A': [(B, 10)],
  'B': [(A, 10), (C, 20)],
  'C': [(B, 20), (D, 15)],
  'D': [(B, 15), (C, 15)],
  'E': [(D, 9)], 
}

至于带多维权重等,可以通过新建一个类、命名元组等方式来实现。它的实现原理是相似的,不多赘述。

这里补充一下,如果可以保证ABCDE对象是可哈希的(hash),那么可以用作字典键,像这样:

vertex = {
  A: [(B, 10)],
  B: [(A, 10), (C, 20)],
  C: [(B, 20), (D, 15)],
  D: [(B, 15), (C, 15)],
  E: [(D, 9)], 
}

也就是直接使用对象的“引用”,不需要转换为字符串。

邻接表是重点

邻接矩阵的优点是简单、容易表示权重、检查两个顶点之间是否有边更快、更容易实现某些图算法等。

但实际生活当中,我们用得更多的是邻接表。因为:

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

推荐阅读更多精彩内容