首页 > 其他 > 详细

二叉树之字形遍历

时间:2021-03-04 23:13:16      阅读:37      评论:0      收藏:0      [点我收藏+]

给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
例如:
给定的二叉树是{3,9,20,#,#,15,7},
技术分享图片

该二叉树之字形层序遍历的结果是

[

[3],

[20,9],

[15,7]

]

代码实现

? 通过使用两个栈来实现之字形遍历,第一个栈存储从左到右的遍历的层,第二个栈存储从右到左遍历的层。其中第一个栈出栈的元素,再将其直系子节点存储栈中时,先存左孩子再存右孩子,这样它们的孩子那一层的元素出栈的时候就是从右到左了;第二个栈中的元素出栈的时候,存储它们的孩子节点的时候,先存右孩子,再存左孩子,这样就保证了它们的孩子层的元素出栈的时候是从左往右了。

? 这两个栈中每次只存储一层的元素,将一层的元素全部出栈完毕之后,才会对下一层的元素进行出栈,这就保证能让同一层的元素存储到一个列表中了。

public ArrayList<ArrayList<Integer>> zigzagLevelOrder (TreeNode root) {

    ArrayList<ArrayList<Integer>> resList = new ArrayList<>();
    if (root == null){
        return resList;
    }

    Stack<TreeNode> first = new Stack<>(); //从左往右
    Stack<TreeNode> second = new Stack<>(); //从右往左

    first.push(root);
    int i = 1;
    while (!first.isEmpty() || !second.isEmpty()){

        ArrayList<Integer> level = new ArrayList<>();
        if (i % 2 != 0){

            while (!first.isEmpty()){

                TreeNode node = first.pop();
                level.add(node.val);
                if (node.left != null){
                    second.push(node.left);
                }

                if (node.right != null){
                    second.push(node.right);
                }
            }
        }else {

            while (!second.isEmpty()){

                TreeNode node = second.pop();
                level.add(node.val);

                if (node.right!=null){

                    first.push(node.right);
                }

                if (node.left != null){
                    first.push(node.left);
                }
            }
        }
        i++;
        resList.add(level);
    }
    return resList;
}

二叉树之字形遍历

原文:https://www.cnblogs.com/yxym2016/p/14483114.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!