32 - I. 从上到下打印二叉树&&32 - II. 从上到下打印二叉树 II&&32 - III. 从上到下打印二叉树 III

题目-从上到下打印二叉树

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

例如:
给定二叉树: [3,9,20,null,null,15,7],
返回：
[3,9,20,15,7]

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int[] levelOrder(TreeNode root) {
        if(root==null){
            return new int[0];
        }

        Deque<TreeNode> queue=new LinkedList<>();
        List<Integer> list=new ArrayList<>();
        queue.add(root);
        
        while(!queue.isEmpty()){
            TreeNode node=queue.remove();
            list.add(node.val);

            if(node.left!=null){
                queue.add(node.left);
            }
            if(node.right!=null){
                queue.add(node.right);
            }
        }

        int[] res=new int[list.size()];
        for(int i=0;i<res.length;i++){
            res[i]=list.get(i);
        }
        return res;
    }
}

复杂度

时间复杂度O(n):n个二叉树的节点个数
空间复杂度O(n) :list

题目-从上到下打印二叉树 II

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

思路

• 层序遍历
• 栈的大小就是每一层节点(上一层取完就是下一层全部入栈之时)

/**
 * 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) {
        List<List<Integer>> list=new ArrayList<>();
        Deque<TreeNode> queue=new LinkedList<>();
        if(root==null){
            return list;
        }
        queue.add(root);

        while(!queue.isEmpty()){
            List<Integer> path=new ArrayList<>();
            //上一层取完时也是下一层加入完，所以栈的大小就是每一层节点的个数
            for(int i=queue.size();i>0;i--){
                TreeNode node=queue.remove();
                path.add(node.val);
                if(node.left!=null){
                    queue.add(node.left);
                }
                if(node.right!=null){
                    queue.add(node.right);
                }
            }
            list.add(path);
        }

        return list;
    }
}

题目-从上到下打印二叉树 III

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：
[
[3],
[20,9],
[15,7]
]

/**
 * 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) {
    
        Deque<TreeNode> queue=new LinkedList<>();
        List<List<Integer>> list=new ArrayList<>();
        if(root!=null){
            queue.add(root);
        }
    
        while(!queue.isEmpty()){
            
            List<Integer> path=new ArrayList<>();

            for(int i=queue.size();i>0;i--){
                TreeNode node=queue.remove();
                path.add(node.val);

                if(node.left!=null){
                    queue.add(node.left);
                }

                if(node.right!=null){
                    queue.add(node.right);
                }
            }
            if(list.size()%2==1){
                Collections.reverse(path);
            }
            list.add(path);
        }
        return list;
    }
}

补充

时间复杂度O(n) :遍历二叉树的所有节点
空间复杂度O(n) ：满二叉树，最多有n/2的节点存在于队列