10.之字形打印二叉树

题目:把二叉树按之字形层序遍历,每层单独保存,最后返回。

public ArrayList<ArrayList> zigzagLevelOrder (TreeNode root)

思路:基本思路是层序遍历每层单独保存。这是用size固定,然后把每次size的元素放到一层中保存。
不同的是有正向的,有反向的。用一个,来实现反向的。(就是先放到栈里,再拿出来保存)。
用栈,不会增加时间和空间的复杂度。(时间上语句没有嵌套,而且需要使用队列)

	 /*
    隔行打印层序遍历,用size保存
    之字形,需要反过来的就放栈里面,复杂度都不会变

    还有用arrayList.add(0,item)来实现反向插入,但是需要移动时间复杂度高
    就不用这个了。就用栈来进行反向操作。
     */
    public ArrayList<ArrayList<Integer>> zigzagLevelOrder (TreeNode root) {
        LinkedList<TreeNode> queue = new LinkedList<>();
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        //之字形需要反过来的时候用
        Stack<Integer> stack = new Stack<>();
        if(root == null) return res;
        queue.add(root);
        //是否需要反过来(是否需要中间放到栈里)
        boolean inverted = false;
        while(!queue.isEmpty()){
            //每次resList都是新的
            ArrayList<Integer> resList = new ArrayList<>();
            //size要提前固定下来
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode temp = queue.poll();
                if(inverted){
                    stack.push(temp.val);
                }else{
                    resList.add(temp.val);
                }
                if(temp.left != null) queue.add(temp.left);
                if(temp.right != null) queue.add(temp.right);
            }
            if(inverted){
                while(!stack.isEmpty()){
                    resList.add(stack.pop());
                }
                inverted = false;
            }else{
                inverted = true;
            }
            res.add(resList);
        }
        return res;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值