LeetCode 103. Binary Tree Zigzag Level Order Traversal

问题描述

  • Given a binary tree, return the zigzag level order traversal of its nodes’ values. (ie, from left to right, then right to left for the next level and alternate between).
  • Example:
    这里写图片描述
  • 地址

问题分析

代码实现

  • 双栈法
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        Stack<TreeNode> left2Right = new Stack<>();
        Stack<TreeNode> right2Left = new Stack<>();
        boolean order = true;
        List<List<Integer>> res = new ArrayList<>();
        left2Right.push(root);
        while (!left2Right.isEmpty() || ! right2Left.isEmpty()) {
            if (order) {
                List<Integer> list = new ArrayList<>();
                while (! left2Right.isEmpty()) {
                    TreeNode popNode = left2Right.pop();
                    list.add(popNode.val);
                    if (popNode.left != null) {
                        right2Left.push(popNode.left);
                    }
                    if (popNode.right != null) {
                        right2Left.push(popNode.right);
                    }
                }
                res.add(list);
            }else {
                List<Integer> list = new ArrayList<>();
                while (! right2Left.isEmpty()) {
                    TreeNode popNode = right2Left.pop();
                    list.add(popNode.val);
                    if (popNode.right != null) {
                        left2Right.push(popNode.right);
                    }
                    if (popNode.left != null) {
                        left2Right.push(popNode.left);
                    }
                }
                res.add(list);
            }
            order = ! order;
        }
        return res;
    }
  • BFS + LinkedList
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        LinkedList<TreeNode> queue = new LinkedList<>();
        boolean order = true;
        List<List<Integer>> res = new ArrayList<>();
        queue.add(root);
        while (! queue.isEmpty()) {
            int levelSize = queue.size();
            LinkedList<Integer> list = new LinkedList<>();
            for (int i = 0; i < levelSize; i++) {
                TreeNode popNode = queue.remove();
                //根据order来确定插入位置
                if (order) {
                    list.addLast(popNode.val);
                }else {
                    list.addFirst(popNode.val);
                }
                //order? list.addLast(popNode.val) : list.addFirst(popNode.val);
                if (popNode.left != null) {
                    queue.add(popNode.left);
                }
                if (popNode.right != null) {
                    queue.add(popNode.right);
                }
            }
            res.add(list);
            order = ! order;
        }
        return res;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值