zoukankan      html  css  js  c++  java
  • 牛客(60)把二叉树打印成多行

        //    题目描述
    //    从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
        public class TreeNode {
            int val = 0;
            TreeNode left = null;
            TreeNode right = null;
    
            public TreeNode(int val) {
                this.val = val;
            }
        }
    
        ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
            ArrayList<ArrayList<Integer>> arrayLists = new ArrayList<ArrayList<Integer>>();
            if (pRoot == null) {
                return arrayLists;
            }
            Queue<TreeNode> stack1 = new LinkedList<TreeNode>();
            Queue<TreeNode> stack2 = new LinkedList<TreeNode>();
            stack1.add(pRoot);
            while (!stack1.isEmpty() || !stack2.isEmpty()) {
                if (!stack1.isEmpty()) {
                    ArrayList<Integer> arrayList = new ArrayList<Integer>();
                    while (!stack1.isEmpty()) {
                        TreeNode node = stack1.poll();
                        arrayList.add(node.val);
                        if (node.left != null) {
                            stack2.add(node.left);
                        }
                        if (node.right != null) {
                            stack2.add(node.right);
                        }
                    }
                    arrayLists.add(arrayList);
                } else if (!stack2.isEmpty()) {
                    ArrayList<Integer> arrayList = new ArrayList<Integer>();
                    while (!stack2.isEmpty()) {
                        TreeNode node = stack2.poll();
                        arrayList.add(node.val);
                        if (node.left != null) {
                            stack1.add(node.left);
                        }
                        if (node.right != null) {
                            stack1.add(node.right);
                        }
                    }
                    arrayLists.add(arrayList);
                }
            }
            return arrayLists;
        }
  • 相关阅读:
    设计模式-状态模式(25)
    设计模式-访问者模式(24)
    设计模式-观察者模式(22)
    设计模式-中介者模式(21)
    设计模式-行为型模式小结(20)
    设计模式-迭代器模式(19)
    Tomcat安装
    MySQL单表查询
    MySQL表操作
    MySQL表的完整性约束
  • 原文地址:https://www.cnblogs.com/kaibing/p/9109988.html
Copyright © 2011-2022 走看看