之字形遍历二叉树——你为何这么浪

说到二叉树遍历,脑海中立刻想到的就是深度优先遍历和广度优先遍历,这两种方式相信大家都驾轻就熟了,就不再过多累赘。

今天和大家分享的是之字形遍历二叉树。

问题描述

请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。

解题思路

设两个栈,s2存放奇数层,s1存放偶数层

遍历s2节点的同时按照左子树、右子树的顺序加入s1,

遍历s1节点的同时按照右子树、左子树的顺序加入s2

示例代码

    public void loop(TreeNode root) {
        if (root == null) return;

        ArrayList<TreeNode> res = new ArrayList<>();

        Stack<TreeNode> stack1 = new Stack<>();
        Stack<TreeNode> stack2 = new Stack<>();
        boolean left2Right = true;

        stack1.push(root);

        while (!stack1.isEmpty() || !stack2.isEmpty()) {
            // 从左向右
            if (left2Right) {

                while (!stack1.isEmpty()) {
                    TreeNode node = stack1.pop();
                    res.add(node);
                    if (node.left != null) {
                        stack2.push(node.left);
                    }
                    if (node.right != null) {
                        stack2.push(node.right);
                    }
                }
            // 从右向左
            } else {

                while (!stack2.isEmpty()) {
                    TreeNode node = stack2.pop();
                    res.add(node);
                    if (node.right != null) {
                        stack1.push(node.right);
                    }
                    if (node.left != null) {
                        stack1.push(node.left);
                    }
                }

            }
            left2Right = !left2Right;
        }

        for (TreeNode node : res) {
            System.out.println("" + node.value);
        }
    }
复制代码

转载于:https://juejin.im/post/5c999c83f265da610d0cd266

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值