Day53 二叉树的锯齿形层序遍历

最新推荐文章于 2021-04-28 23:29:19 发布

微光长明

最新推荐文章于 2021-04-28 23:29:19 发布

阅读量95

点赞数

分类专栏： LeetCode刷题集文章标签：算法 leetcode

本文链接：https://blog.csdn.net/walkmanfy/article/details/115035818

版权

LeetCode刷题集专栏收录该内容

90 篇文章 0 订阅

订阅专栏

给定一个二叉树，返回其节点值的锯齿形层序遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）

https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal/

示例1：

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

3
/ \
9 20
/ \
15 7
返回锯齿形层序遍历如下：

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

Java解法

思路：

昨天用栈实现时就导致了这种现象
放入取出再放入正好带来了这种效果

package sj.shimmer.algorithm.m3_2021;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

import sj.shimmer.algorithm.TreeNode;

/**
 * Created by SJ on 2021/3/19.
 */

class D53 {
    public static void main(String[] args) {
        System.out.println(zigzagLevelOrder(TreeNode.getInstance(new Integer[]{3, 9, 20, null, null, 15, 7})));
        System.out.println(zigzagLevelOrder(TreeNode.getInstance(new Integer[]{1,2,3,4,5})));

    }
    public static List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> results = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        if (root != null) {
            stack.add(root);
        }
        boolean toRight = true;
        while (!stack.isEmpty()){
            List<Integer> tempList = new ArrayList<>();
            Stack<TreeNode> temp = new Stack<>();
            while (!stack.isEmpty()) {
                TreeNode pop = stack.pop();
                if (pop != null) {
                    tempList.add(pop.val);
                    if (toRight) {
                        temp.add(pop.left);
                        temp.add(pop.right);
                    }else {
                        temp.add(pop.right);
                        temp.add(pop.left);
                    }
                }
            }
            toRight=!toRight;
            if (tempList.size()!=0) {
                results.add(tempList);
                stack = temp;
            }
        }
        return results;
    }
}