数据结构之图节点Vertex

数据结构之图节点Vertex

#!/usr/bin/env python2.7
# -*- conding: utf-8 -*-
# @Time : 2020/10/30 17:34
# @Author : ada
# @file : Vertex.py.py
# @Project: python-script 

class Vertex:
    def __init__(self,key):
        self.id = key
        self.connectedTo = {}
        
        '''节点id属性为节点的key
           节点的connectedTo 字典存放节点的n个连接点，每个连接点有自已的key,值为连接权重
        '''

    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):
        reutrn self.connectedTo[nbr]