剑指offer32 从上到下打印二叉树3

渔舟舟

于 2020-09-17 12:56:59 发布

阅读量126

点赞数

分类专栏：剑指offer 文章标签：二叉树层序遍历之字形

本文链接：https://blog.csdn.net/realYuzhou/article/details/108640149

版权

剑指offer 专栏收录该内容

29 篇文章 0 订阅

订阅专栏

本文介绍如何按照之字形顺序打印二叉树，详细阐述了思路，并提供了实现代码，涉及二叉树的层序遍历和奇数行逆序处理。

摘要由CSDN通过智能技术生成

问题描述

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

思路

第三个题目是在第二个题目的基础上，增加了奇数行逆序的步骤，直接在第二个题目的基础上增加一个判断，逆序就行了。

代码

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if(root == null) return res;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        int level = 0;
        while(!queue.isEmpty()) {
            ArrayList<Integer> temp = new ArrayList<>();
            for (int i = queue.size(); i > 0; i--) {
                TreeNode node = queue.poll();
                if(node.left != null) queue.offer(node.left);
                if(node.right != null) queue.offer(node.right);
                temp.add(node.val);
            }
            if (level % 2 != 0) {
                Collections.reverse(temp);
            }
            res.add(temp);
            level++;
        }
        return res;
    }
}