【树】剑指 Offer 32 - III. 从上到下打印二叉树 III

题目:

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

 

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

3
/ \
9 20
/ \
15 7
返回其层次遍历结果:

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

 

解答:

方法一:

在剑指 Offer 32 - II. 从上到下打印二叉树 II解答的基础上增加一个isOdd变量,当是奇数层时从左到右遍历,否则从右到左遍历。

/**
 * 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) {
        ArrayList<List<Integer>> res = new ArrayList<>();
        if(root == null) return res;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        boolean isOdd = true;
        while(!queue.isEmpty()){
            ArrayList<Integer> curLevel = new ArrayList<>();
            for(int i = queue.size();i>0;i--){
                TreeNode node = queue.poll();
                curLevel.add(node.val);
                if(node.left!=null) queue.add(node.left);
                if(node.right!=null) queue.add(node.right);
            }
            if(isOdd)  res.add(curLevel);
        //偶数层倒叙,还可以用Collections.reverse(tmp);
if(!isOdd){ ArrayList<Integer> tmp = new ArrayList<>(); for(int i = curLevel.size()-1;i>=0;i--){ tmp.add(curLevel.get(i)); } res.add(tmp); } isOdd = !isOdd; } return res; } }

用 Collections.reverse 反转一个 List

https://www.cnblogs.com/ywb2018/p/9922829.html 

 

方法二:层序遍历+双端队列

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        List<List<Integer>> res = new ArrayList<>();
        if(root != null) queue.add(root);
        while(!queue.isEmpty()) {
            LinkedList<Integer> tmp = new LinkedList<>();
            for(int i = queue.size(); i > 0; i--) {
                TreeNode node = queue.poll();
                if(res.size() % 2 == 0) tmp.addLast(node.val); // 偶数层 -> 队列头部
                else tmp.addFirst(node.val); // 奇数层 -> 队列尾部
                if(node.left != null) queue.add(node.left);
                if(node.right != null) queue.add(node.right);
            }
            res.add(tmp);
        }
        return res;
    }
}

 

posted @ 2020-08-26 20:33  3KBLACK  阅读(61)  评论(0)    收藏  举报