栈---二叉树的锯齿形层次遍历

  1. 题目
    给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
  2. 示例
给定二叉树 [3,9,20,null,null,15,7],

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

[
  [3],
  [20,9],
  [15,7]
]
  1. 代码实现
    数据结构:栈;
    currStack:存储当前层的元素;
    nextStack:存储下一层的元素。
    order:boolean类型,控制左,右孩子的加入顺序,实现锯齿形。
    思路:
    a.在弹出currentStack元素时,将该元素的左,右子孩子记录到nextStack中,直到currStack元素全部弹出。(即当前层元素全部输出)(注:左,右孩子的加入顺序由order变量控制)
    b.交换currStack和nextStack,使currStack始终指向当前需要输出的层。
    c.order取反
import java.util.*;

public class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root){
        List<List<Integer>> lists = new ArrayList<>();
        //存储当前层元素
        Stack<TreeNode> currStack = new Stack<>();
        //存储下一层元素
        Stack<TreeNode> nextStack = new Stack<>();
        if(root!=null){
            currStack.add(root);
        }
        //控制锯齿形
        boolean order = true;
        while (!currStack.empty()){
            List<Integer> list = new ArrayList<>();
            //将当前层元素全部弹出
            while (!currStack.empty()){
                TreeNode top = currStack.pop();
                list.add(top.val);
                //将弹出元素的左右孩子记录到nextStack中
                if(order){
                    if(top.left!=null){
                        nextStack.add(top.left);
                    }
                    if(top.right!=null){
                        nextStack.add(top.right);
                    }
                }else{
                    if(top.right!=null){
                        nextStack.add(top.right);
                    }
                    if(top.left!=null){
                        nextStack.add(top.left);
                    }
                }
            }
            //交换currStack和nextStack,使currStack始终指向当前层
            currStack = nextStack;
            nextStack = new Stack<>();
            lists.add(list);
            //order取反
            order =!order;
        }
        return lists;
    }
}

注:常规层次遍历使用队列实现

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值