python中任意数

                          学校
            商学院     法学院    电子学院       计算机学院
                                                       软件        网络       数据
                                                                 网1 网2 网3
1.根据当前节点 获取父节点
2.根据当前节点 获取所有子节点
3.根据当前节点 获取所有祖先节点
4.根据当前节点 获取所有子孙节点
5.根据当前节点 获取所有兄弟节点

class Tree:
    childToParent=None
    parentToChildren=None
    def add(self,parent,child):
        if self.childToParent  is None:  #如果childToParent 是空的话：
            self.childToParent={child:parent} #转集合
        else:
            self.childToParent[child]=parent
        if self.parentToChildren is None:
            self.parentToChildren={parent:[]} #父亲对应 多个孩子
        children=self.parentToChildren.get(parent,[])
        if len(children)==0:
            self.parentToChildren[parent]=children
        children.append(child) #直接追加
    def getParent(self,child):
        return self.childToParent.get(child,'没有父亲的节点')
    def getChildren(self,parent):
        return self.parentToChildren.get(parent,'没下级')
    def getZuXian(self,zisun):
        '''
        思想就是 递归问爹
        1.可以获取上一级 getParent
        2.如果我能够让上一级返回的数据在 执行 getParent 依次执行 直到没有上级终止
        :param zisun:
        :return:
        '''
        parent = self.getParent(zisun)
        if parent is None:
            return []
        zupu= self.getZuXian(parent)
        zupu.append(parent)
        return zupu
    #第四
    def zisun(self,zs,zu):
        '''
        1.得到当前节点的孩子
        2.遍历当前节点的孩子 每一个孩子去得到他们的孩子
        :param zs: 结果
        :param zu: 条件
        :return:
        '''
        children=self.getChildren(zu)
        if children == '没下级':
            return
        for c in children:
            self.zisun(zs,c)
        zs.extend(children)
        #第5
    def getSibling(self,sibling):
        parent=self.getParent(sibling)
        children=self.getChildren(parent)
        children.remove(sibling)
        return  children

if __name__=='__main__':
    tree=Tree()
    tree.add(None,'学校')
    tree.add('学校','商学院 ')
    tree.add('学校','法学院')
    tree.add('学校','电子学院 ')
    tree.add('学校','计算机学院')
    tree.add('计算机学院','软件')
    tree.add('计算机学院','网络')
    tree.add('计算机学院','数据')
    tree.add('网络','网1')
    tree.add('网络','网2')
    tree.add('网络','网3')
#1.根据当前节点获取父节点  网路
    child='网络'
    parent=tree.getParent(child)
    print('{1}的上级是：{0}'.format(parent,child))
#2. 根据当前节点 获取所有子节点
    parent = '计算机学院'
    children = tree.getChildren(parent)
    print('{0}的子级是:{1}'.format(parent,children))
# 3.根据当前节点 获取所有祖先节点 ancestor
    zisun = '网3'
    zuxian=tree.getZuXian(zisun)
    print('{0}的祖先是:{1}'.format(zisun, zuxian))
#4. 根据当前节点 获取所有子孙节点
    zu='学校'
    zs=[]
    tree.zisun(zs,zu)
    print('{0}的子孙是：{1}'.format(zu,zs))
#5. 根据当前节点 获取所有兄弟节点
    sibling='法学院'
    rs = tree.getSibling(sibling)
    print('{0}的兄弟是:{1}'.format(sibling,rs))

运行结果如图：

完毕！