leetcode-107 二叉树的层次遍历

给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)


思路:容易想到对树的层次遍历是通过队列完成的,不过发现python的queue模块并没有提供循环队列,只提供了简单的队列,好处是不会限制队列的大小,难点在于怎么分层次的把节点值存到内层队列中,方法是:先用size获取队列的大小即为该层次上的节点个数,然后一次访问队列的前size个节点,每次访问完一层都要将获取到的队列的值,返回到结果中

下面是python代码:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

from queue import Queue

class Solution:
    
    def levelOrderBottom(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        result = []
        
        #空树,直接返回
        if root == None:
            return result
        
        #非空使用队列解决问题,采用二叉树的层次遍历
        q = Queue()
        q.put(root)
        
        #使用二层循环输出队列
        while not q.empty():
            
            size = q.qsize()
            temp_list = []
            
            #使用size获取每层的队列的大小
            #内层循环获取头节点,并且处理数据,然后将其子节点入队
            while size > 0:
                temp_node = q.get()
                temp_list.append(temp_node.val)
                
                if temp_node.left != None:
                    q.put(temp_node.left)
                if temp_node.right != None:
                    q.put(temp_node.right)
                size -= 1
                
            result.append(temp_list)
            
        result.reverse()
        
        return result

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章