题解 | 按之字形顺序打印二叉树

按之字形顺序打印二叉树

https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> Print (TreeNode pRoot) {
        // write code here
        TreeNode head = pRoot;
        ArrayList<ArrayList<Integer>> res = new ArrayList();
        if(head == null) return res;
        Stack<TreeNode> s1 = new Stack<TreeNode>();
        Stack<TreeNode> s2 = new Stack<TreeNode>();
        s1.push(head);
        while(!s1.isEmpty()||!s2.isEmpty()){
            ArrayList<Integer> temp = new ArrayList<Integer>();
            while(!s1.isEmpty()){
                TreeNode node = s1.pop();
                temp.add(node.val);
                if(node.left!= null){
                    s2.push(node.left);
                }
                if(node.right != null){
                    s2.push(node.right);
                }
            }
            if(temp.size()!= 0) res.add(new ArrayList<Integer>(temp));
            temp.clear();
            while(!s2.isEmpty()){
                TreeNode node = s2.pop();
                temp.add(node.val);
                if(node.right!= null){
                    s1.push(node.right);
                }
                if(node.left!=null){
                    s1.push(node.left);
                }
            }
            if(temp.size()!= 0) res.add(new ArrayList<Integer>(temp));
            temp.clear();
        }
        

        return res;
    }
}

与之前的层序遍历类似 只是需要使用两个栈来倒换方向

全部评论

相关推荐

燕山北:项目写点监控 可视化 高可用 自动化 不要总是写部署过什么,写点能够体现水平的
投了多少份简历才上岸
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务