学习Python有一段时间了,看了几本书,大部分时间都是拿它写一些小的脚本,处理数据,最近用Python实现了几个常见的图论算法。第一部分就是图的存储和建立。

方法一,用矩阵存储。所谓矩阵就是二维数组。Python的二维数组用起来更加随意和简单。写了一个很low的控制台输入建立图的方法。一时间也没想起其他好的方法。

用二维数组存储,对算法的实现很有帮助。

return graph

先输入节点数n,再输入边的数量m。注意我们建立的是无向图。然后先构建一个N x N的二维数组并且全部赋值为inf。意为边的大小都是无限大。接下来输入m条边,边的表示用a b w 表示(ab)为起点,w为权重。

第二种方法,用边级的形式。