面试题32-III.从上到下打印二叉树III

题目

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

示例

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

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

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

代码

和前一题一样,只要加一行判断
从上到下打印二叉树,又称为二叉树的广度优先搜索(BFS),BFS通常借助队列来实现。
算法流程

  1. 特例处理: 当根节点为空,则返回空列表 [] ;
  2. 初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root]
  3. BFS 循环: 当队列 queue 为空时跳出;
    1. 新建一个临时列表 tmp ,用于存储当前层打印结果;
    2. 当前层打印循环: 循环次数为当前层节点数(即队列 queue 长度);
      1. 出队: 队首元素出队,记为 node
      2. 打印:node.val 添加至 tmp 尾部;
      3. 添加子节点:node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue
    3. 将当前层结果 tmp 添加入 res
  4. 返回值: 返回打印结果列表 res 即可。
/**
 * 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 new ArrayList<>();
        //队列
        Deque<TreeNode> queue = new ArrayDeque<>();
        queue.addLast(root);//将root加入到队列
        //返回值
        List<List<Integer>> list = new ArrayList<>();
        
        while (queue.size() != 0) {
        	//内层的List
            List<Integer> innerList = new ArrayList<>();
            //前一个循环结束后,队列的长度就是树的下一层节点个数
            for (int i = queue.size(); i > 0; i--) {
                TreeNode node = queue.removeFirst();
                innerList.add(node.val);
                if (node.left != null) queue.addLast(node.left);
                if (node.right != null) queue.addLast(node.right);
            }
			
			//size和与1为1则表示size为奇数,代表当前行为偶数行,需要翻转输出
			if ((list.size() & 1) == 1) Collections.reverse(innerList);
            list.add(innerList);
        }

        return list;
    }
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof

你可能感兴趣的:(剑指offer,LeetCode)