二叉树层次遍历

class TreeNode(object):
     def __init__(self, x):
         self.val = x
         self.left = None
         self.right = None
class Solution(object):
    # 分层输出
    def levelorder_1(self, root):
        q = [[root]]
        for level in q:
            print(level)
            record = []
            for node in level:
                if node.left: record.append(node.left)
                if node.right: record.append(node.right)
            if record: q.append(record)
        res=[]
        for i in q:
            tem=[]
            for j in i:
                tem.append(j.val)
            res.append(tem)
        print(res)
        return res
    # 按顺序输出
    def levelorder_2(self, root):
        deque = [root]
        res=[]
        while deque:
            t=deque.pop(0)
            res.append(t.val)
            if t.left:
                deque.append(t.left)
            if t.right:
                deque.append(t.right)
        print(res)
        return res
if __name__ == '__main__':
    S = Solution()
    l1 = TreeNode(1)
    l2 = TreeNode(2)
    l3 = TreeNode(3)
    l4 = TreeNode(4)
    l5 = TreeNode(5)
    l6 = TreeNode(6)
    l7 = TreeNode(7)
    root = l1
    l1.left = l2
    l1.right = l3
    l2.left = l4
    l2.right = l5
    l3.left = l6
    l3.right = l7
    z = S.levelorder_2(root)

分别代表分层输出以及利用队列按顺序输出

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值