leetcode樹之從上到下打印二叉樹

本文主要記錄一下leetcode樹之上到下打印二叉樹node

題目

從上到下按層打印二叉樹,同一層的節點按從左到右的順序打印,每一層打印到一行。

 

例如:
給定二叉樹: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其層次遍歷結果:

[
  [3],
  [9,20],
  [15,7]
]

 

提示:

    節點總數 <= 1000

注意:本題與主站 102 題相同:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/

來源:力扣(LeetCode)
連接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof
著做權歸領釦網絡全部。商業轉載請聯繫官方受權,非商業轉載請註明出處。

題解

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) {
            return Collections.emptyList();
        }

        Queue queue = new LinkedList();
        queue.offer(root);
        List<List<Integer>> result = new ArrayList();
        while (!queue.isEmpty()) {
            List<Integer> list = new ArrayList<>();
                int length = queue.size();
                for (int i=0; i< length; i++) {
                    TreeNode node = (TreeNode)queue.poll();
                    if (node != null) {
                        list.add(node.val);
                        queue.offer(node.left);
                        queue.offer(node.right);
                    }
                }
            if (!list.isEmpty()){
                result.add(list);  
            }  
        }

        return result;
    }
}

小結

這是二叉樹層次遍歷的一個變種,區別在於每次poll的以前須要先記錄下當前queue的size,即當前層的節點個數,而後按這個size去pull。網絡

doc

相關文章
相關標籤/搜索