zoukankan      html  css  js  c++  java
  • 二叉树之字形遍历

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

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

    [

    [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;
    }
    
  • 相关阅读:
    vagrant 的安装与使用
    vagrant 的安装与使用
    rz、sz (上传下载)命令参数的解释
    rz、sz (上传下载)命令参数的解释
    TensorFlow 学习(十五)—— tensorflow.python.platform
    音乐的作曲形式
    vc中edit控件使用总结
    引用 LPSTR、LPCSTR、LPTSTR、LPCTSTR、LPWSTR及LPCWSTR的意义及区别
    编译原理三大经典书籍(龙书 虎书 鲸书)
    VS2003与Win7的兼容性问题
  • 原文地址:https://www.cnblogs.com/yxym2016/p/14483114.html
Copyright © 2011-2022 走看看