Python中的图形表示与邻接列表

时间:2014-10-29 10:24:59

标签: python

我在Python中找到了以下Graph实现:

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

    def addNeighbor(self,nbr,weight=0):
        self.connectedTo[nbr] = weight

    def getConnections(self):
        return self.connectedTo.keys()

    def getId(self):
        return self.id

    def getWeight(self,nbr):
        return self.connectedTo[nbr]

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 addEdge(self,f,t,cost=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], cost)

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

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


def main():
    g=Graph()
    for i in range(6):
        g.addVertex(i)
    g.addEdge(0,1,5)
    g.addEdge(1,5,4)
    g.addEdge(5,3,6)
    g.addEdge(3,4,5)
    for v in g:
        for w in v.getConnections():
            print v.getId(),",",w.getId(),",",v.getWeight(w)

if __name__=="__main__":
    main()

程序运行正常,但我想知道作者是否放了一些冗余部分,比如函数addEdge()addVertex()

对于我看到的内容,addEdge有一个从不使用的赋值:

nv=self.addVertex(t)

也是说:

if t not in self.vertList:
    nv=self.addVertex(t)

在同一方法中没有做任何事情,并且根本不使用变量nv。对于我所看到的,这是有向图的实现,所以如果我想编写一个非有向图,它就足够了:

self.vertList[t].addNeighbor(self.vertList[f], cost)

所以我也可以摆脱addVertex函数中的返回newVertex。我做了那些修改,程序仍然很好。当我重用这个类时,这些修改是否会很好并且不会带来任何奇怪的行为?

最后一个问题,在部分中说:

for v in g:
    for w in v.getConnections():
        print v.getId(),",",w.getId(),",",v.getWeight(w)

它似乎只迭代G字典中具有值而不是空值的那些值(我的意思是那些只有一个键),是这样的吗?

1 个答案:

答案 0 :(得分:1)

是的,我认为nv=是多余的,但self.addVertex(f)是必要的,因为您需要添加顶点。
添加self.vertList[t].addNeighbor(self.vertList[f], cost)将生成非有向图,为真。

  

它似乎只迭代G字典中的那些值   有价值而不是空的(我的意思是那些有价值的   只有一把钥匙),是这样的吗?

不,它将迭代所有顶点,但只有那些有连接(边缘)的顶点才会生成输出。