python应用——邻接Graph类实现公交换乘系统

最新推荐文章于 2023-08-11 15:29:05 发布

北落师门XY

最新推荐文章于 2023-08-11 15:29:05 发布

阅读量1.1k

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/weixin_41819299/article/details/80840155

版权

python 专栏收录该内容

24 篇文章 2 订阅

订阅专栏

一、题目要求

参考提供论文，用书上方式（邻接Graph类或压缩方式）实现问题的解决，将论文中的4路公交换成10路公交，自设连通方式，求解任意公交站点能否换乘？

二、思路

用10*10的矩阵表示公交站之间能否换乘，在原始矩阵中用1代表两路公交经过1次即可换乘，0代表两路公交无共同停靠点。不断做矩阵乘法，比较经过换乘后的矩阵与换乘前有何不同，与原始论文用“1*代表换乘后可连通”的设定不同的是：我采用2,3...代表需要换乘的次数，在若干次换乘后，矩阵不再改变，若此时矩阵中还有0，则该处代表的两个公交站点不可换乘。

三、调用函数及类

# -*-coding:gbk-*-
from numpy import *

class GraphError(Exception):
    pass

class Graph: #用邻接矩阵实现基本图类
    def __init__(self, mat, unconn = 0):#两参数：原始矩阵，初始无连接点默认值0
        vnum = len(mat)#公交路线数目
        for x in mat:
            if len(x) != vnum: # 检查是否为方阵
                raise ValueError("Argument for 'Graph'.")
        self._mat = [mat[i][:] for i in range(vnum)] # 做矩阵的拷贝
        self._unconn = unconn
        self._vnum = vnum#共三参数：矩阵，无关联特殊值，公交路线数

    def _invalid(self, v): # 检查 v 是否超出矩阵范围
        return 0 > v or v >= self._vnum

    def vertex_num(self):#获得顶点数
        return self._vnum

    def add_vertex(self):#暂时不支持顶点数添加
        raise GraphError("Adj-Matrix does not support 'add_vertex'.")

    def add_edge(self, vi, vj, val=1):#将从v1到v2的边加入这个图，权重数va1默认为1
        if self._invalid(vi) or self._invalid(vj):
            raise GraphError(str(vi) + ' or ' + str(vj) + " is not a valid vertex.")
        self._mat[vi][vj] = val

    def get_edge(self, vi, vj):#得到边的权重
        if self._invalid(vi) or self._invalid(vj):
            raise GraphError(str(vi) + ' or ' + str(vj) + " is not a valid vertex.")
        return self._mat[vi][vj]

    def out_edges(self, vi):#返回vi行连接节点表
        if self._invalid(vi):
            raise GraphError(str(vi) + " is not a valid vertex.")
        return self._out_edges(self._mat[vi], self._unconn)

    @staticmethod
    def _out_edges(row, unconn):#构造某行连接节点表[(节点，权重)。。。]
        edges = []
        for i in range(len(row)):
            if row[i] != unconn:
                edges.append((i, row[i]))
        return edges

    def __str__(self):#转换为字符串形式的输出：矩阵+无关联元素 map将操作运用到序列的每一个元素
        return "[\n" + "\n".join(map(str, self.mat)) + "\n]\n" \
               + "Unconnected: " + str(self.unconn)

    def findn(self):#自定义寻找换乘次数函数
        matpast=[self._mat[i][:] for i in range(self._vnum)]
        cishu=2
        panduanwancheng = 0
        while panduanwancheng==0 :
            matnow = zeros((self._vnum, self._vnum))
            panduanwancheng = 1
            for i in range(self._vnum):#注意range是开区间的
                for j in range(self._vnum):
                    for k in range(self._vnum):
                        matnow[i][j]+=self._mat[i][k]*matpast[k][j]
                    if matnow[i][j]!=0 and matpast[i][j]==0:
                        matnow[i][j]=cishu#需要换乘次数
                        panduanwancheng = 0
                    if matpast[i][j] != 0:
                        matnow[i][j]=matpast[i][j]
            cishu+=1
            matpast = [matnow[b][:] for b in range(self._vnum)]

        return matnow

四、主函数

def main():
    print("************************************验证自定义函数是否正确")
    mat1=[[1,1,0,0],
          [1,1,0,1],
          [1,0,1,0],
          [0,0,0,1]]
    ceshimat=Graph(mat1)
    yanzheng=ceshimat.findn()
    print(yanzheng)
    print("************************************十路公交")
    mat2=[[1,1,0,1,0,1,0,1,1,0],
          [1,1,0,0,1,1,0,1,0,1],
          [0,0,1,1,0,0,1,1,0,1],
          [0,0,1,1,0,0,1,1,0,1],
          [0,0,0,1,1,1,0,1,0,1],
          [0,0,0,1,1,1,0,1,1,1],
          [1,1,1,1,0,0,1,0,0,0],
          [0,0,0,0,1,1,0,1,0,1],
          [0,0,0,0,1,1,0,1,1,1],
          [1,1,0,0,1,1,0,1,1,1]]
    mymat = Graph(mat2)
    answer = mymat.findn()
    print(answer)
    if 0 not in answer:
        print('任意两站可连通')

if __name__=='__main__':
    main()