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

最新推荐文章于 2024-11-01 14:48:20 发布

Sumxmq

最新推荐文章于 2024-11-01 14:48:20 发布

阅读量48

点赞数

分类专栏： LeetCode-剑指Offer 文章标签： java 算法数据结构

本文链接：https://blog.csdn.net/m0_51957892/article/details/124772942

版权

LeetCode-剑指Offer 专栏收录该内容

35 篇文章 0 订阅

订阅专栏

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

题目描述

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

示例

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

返回其层次遍历结果：

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

数据限制

节点总数 <= 1000

思路

和上两题类似，该题只要最后将得到的列表逐一反转，然后输出即可

时间复杂度：O(n)

空间复杂度：O(n)

代码：

import java.util.*;

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    List list = new ArrayList();
    ArrayList<Integer> lis[] = new ArrayList[2005];
    int top = 0;

    public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null){
            return list;
        }

        for (int i = 0; i <= 2000; i++){
            lis[i] = new ArrayList();
        }

        dfs(root, 1);
        for (int i = 1; i <= top; i++){
            if (i % 2 == 0){
                List ans = new ArrayList<>();
                for (int j = lis[i].size() - 1; j >= 0; j--){
                    ans.add(lis[i].get(j));
                }
                list.add(ans);                
            }else{
                list.add(lis[i]);
            }
        }
        return list;
    }

    void dfs(TreeNode p, int x){
        top = Math.max(top, x);
        lis[x].add(p.val);
        if (p.left != null){
            dfs(p.left, x + 1);
        }
        if (p.right != null){
            dfs(p.right, x + 1);
        }
    }
}