JZ77 按之字形顺序打印二叉树

描述

给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)

数据范围:0 \le n \le 15000≤n≤1500,树上每个节点的val满足 |val| <= 100∣val∣<=100
要求:空间复杂度:O(n)O(n),时间复杂度:O(n)O(n)

例如:
给定的二叉树是{1,2,3,#,#,4,5}

JZ77 按之字形顺序打印二叉树


该二叉树之字形层序遍历的结果是 [[1],[3,2],[4,5]]

思路

自己未想出来,看的代码随想录的二叉树层次遍历写出来的。

1.先将根节点填入队列中,用栈存储其顺序到第三层就会乱。

2.然后将其弹出队列,弹出的同时将其左、右孩子存入到队列中,然后此时队列的长度为当前层元素的个数(此时队列里的元素必然大于等于之前得到的队列长度)。弹出一个元素就将其左、右孩子加入到队列中。

3.遍历的将当前队列中每层的元素弹出,存入到集合中。此时要判断是否是倒序存储了,行数从1开始,奇数行为正序,偶数行为倒序,因此需要一个行数计数器。

4.直到队列为空结束以上操作。

代码

import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        if(pRoot == null){
            return new ArrayList<>();
        }
        Queue<TreeNode> que = new LinkedList<TreeNode>();
        ArrayList<ArrayList<Integer>> resList = new ArrayList<>();
        
        que.add(pRoot);
        int levelNumber = 0;
        while(!que.isEmpty()){
            levelNumber++;
            int length = que.size();
            //存放每层元素的集合
            ArrayList<Integer> item = new ArrayList<>();
            while(length > 0){
                TreeNode cur = que.poll();
                 item.add(cur.val);
                if(cur.left != null){
                    que.add(cur.left);
                }
                if(cur.right != null){
                    que.add(cur.right);
                }
                length--;
            }
            if(levelNumber%2==0){
                    //逆序
                    Collections.reverse(item);
                    resList.add(item);
                }else{
                    //正序
                    resList.add(item);
                }
           
              
        }
        return resList;
    }

}

上一篇:637. 二叉树的层平均值


下一篇:107. 二叉树的层序遍历 II