ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

邻接表的python实现与深度优先搜索

2022-01-02 12:57:59  阅读:231  来源: 互联网

标签:__ vertList 优先 python self 邻接 visited stack def


邻接表的python实现与深度优先搜索

Vertex类

class Vertex:
    def __init__(self,key):
        self.id = key
        self.connectedTo = {}

    #从这个顶点添加一个连接到另一个
    def addNeighbor(self,nbr,weight = 0):
        self.connectedTo[nbr] = weight

    def __str__(self):
        return str(self.id) + 'connectedTo' + str([x.id for x in self.connectedTo])

    #返回邻接表中的所有的项点
    def getConnections(self):
        return  self.connectedTo.keys()

    def getId(self):
        return self.id

    #返回从这个顶点到作为参数顶点的边的权重
    def getweight(self,nbr):
        return  self.connectedTo[nbr]

Graph类

class Graph:
    def __init__(self):
        self.vertList = {}
        self.numVertices = 0

    def addVertex(self,key):
        self.numVertices = self.numVertices + 1
        newVertex = Vertex(key)
        self.vertList[key] = newVertex
        return  newVertex

    def getVertex(self,n):
        if n in self.vertList:
            return  self.vertList[n]
        else:
            return  None

    def __contains__(self, n):
        return  n in self.vertList

    def addEdge(self,f,t,const = 0):
        if f not in self.vertList:
            nv = self.addVertex(f)
        if t not  in self.vertList:
            nv = self.addVertex(t)
        self.vertList[f].addNeighbor(self.vertList[t],const)

    def getVertices(self):
        return  self.vertList.keys()

    def __iter__(self):
        return  iter(self.vertList.values())

基于邻接表的深度优先搜索(python实现)

visited = []
start = 0   #假设以0为起点
stack = []
stack.append(start)
while len(stack)>0:
    n = stack[-1]
    stack.pop()
    if n in visited:
        continue
    print(n," is visited")
    v = g.getVertex(n)
    for neighbor in list(v.getConnections()):
        stack.append(neighbor.getId())
    visited.append(n)

测试

g = Graph()
for i in range(6):
    g.addVertex(i)
g.addEdge(0,1,5)
g.addEdge(0,5,2)
g.addEdge(1,2,4)
g.addEdge(2,3,9)
g.addEdge(3,4,7)
g.addEdge(3,5,3)
g.addEdge(4,0,1)
g.addEdge(5,4,8)
g.addEdge(5,2,1)

visited = []
start = 0 
stack = []
stack.append(start)
while len(stack)>0:
    n = stack[-1]
    stack.pop()
    if n in visited:
        continue
    print(n," is visited")
    v = g.getVertex(n)
    for neighbor in list(v.getConnections()):
        stack.append(neighbor.getId())
    visited.append(n)

输出:

0 is visited
5 is visited
2 is visited
3 is visited
4 is visited
1 is visited

标签:__,vertList,优先,python,self,邻接,visited,stack,def
来源: https://blog.csdn.net/weixin_52925782/article/details/122275456

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有