前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >之字形遍历二叉树——你为何这么浪

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

作者头像
蜻蜓队长
发布2019-04-26 13:47:21
5370
发布2019-04-26 13:47:21
举报
文章被收录于专栏:Android机动车

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

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

问题描述

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

解题思路

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

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

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

示例代码

代码语言:javascript
复制
    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);
        }
    }
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2019-03-26,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Android机动车 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 问题描述
  • 解题思路
  • 示例代码
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档