我在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字典中具有值而不是空值的那些值(我的意思是那些只有一个键),是这样的吗?
答案 0 :(得分:1)
是的,我认为nv=
是多余的,但self.addVertex(f)
是必要的,因为您需要添加顶点。
添加self.vertList[t].addNeighbor(self.vertList[f], cost)
将生成非有向图,为真。
它似乎只迭代G字典中的那些值 有价值而不是空的(我的意思是那些有价值的 只有一把钥匙),是这样的吗?
不,它将迭代所有顶点,但只有那些有连接(边缘)的顶点才会生成输出。