数据结构与算法 | 树的广度优先遍历(完全二叉树)

# CY3761 | 2021-12-21 17:18

"""
02.树的广度优先遍历

完全二叉树的 add
"""


def getId(o):
    if o is None:
        return None
    
    return hex(id(o))[7:]


# 节点类
class Node:
    def __init__(self, item):
        self.item = item  # 节点值
        self.lPoi = None  # 左指针域
        self.rPoi = None  # 右指针域


# 树类
class Tree:
    def __init__(self):
        self.root = None  # 根节点
    
    def add(self, item):
        """
        添加节点 往树添加一个节点 并保证添加后的树是一个完全二叉树
        从上往下 从左往右 遍历树 找到树的空位
        把节点添加进去
        :param item: 添加节点的值
        :return:
        """
        
        print('-' * 80)
        
        oNode = Node(item)  # 创建节点
        
        if not self.root:  # 判断 root 是否为空
            print(item)
            self.root = oNode
            return
        
        items = [self.root]  # 每一层的节点进行存储
        
        while True:
            aItem = items.pop(0)  # 弹出第一个
            
            print(item, getId(aItem), aItem.item, not aItem.lPoi, not aItem.rPoi)
            
            if not aItem.lPoi:  # 先找左边是否为空
                aItem.lPoi = oNode
                return
            
            elif not aItem.rPoi:  # 再找右边是否为空
                aItem.rPoi = oNode
                return
            
            # 如果左右都非空 则 加入数据
            items.extend((aItem.lPoi, aItem.rPoi))  # pop 从后面弹出


if __name__ == '__main__':
    oT = Tree()
    
    for _ in range(10):
        oT.add(_)
    
    """
        _:0 | n0 | root:n0 | n0=None,None
        
        _:1 | n1 | [r] | p-n0 | n0.0=n1
        _:2 | n2 | [r] | p-n0 | n0.1=n2 | n0=n1,n2
        
        _:3 | n3 | [r] | p-n0 n0=n1,n2 [n0=n1,n2] | p-n1 | n1.0=n3
        _:4 | n4 | [r] | p-n0 n0=n1,n2 [n0=n1,n2] | p-n1 | n1.1=n4 | n1=n3,n4
        
        _:5 | n5 | [r] | p-n0 n0=n1,n2 [n0=n1,n2] | p-n1 n1=n3,n4 [n2,n1=n3,n4] | p-n2 n2.0=n5
        _:6 | n6 | [r] | p-n0 n0=n1,n2 [n0=n1,n2] | p-n1 n1=n3,n4 [n2,n1=n3,n4] | p-n2 n2.1=n6 n2=n5,n6
        
        _:7 | n7 | [r] | p-n0 n0=n1,n2 [n0=n1,n2] | p-n1 n1=n3,n4 [n2,n1=n3,n4] | p-n2 n2=n5,n6 [n3,n4,n5,n6] p-n3 n3.0=None,n7

    """