32 从上到下打印二叉树

最新推荐文章于 2023-08-02 19:15:28 发布

原创最新推荐文章于 2023-08-02 19:15:28 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

剑指offer 专栏收录该内容

22 篇文章

订阅专栏

题目一：不分行从上到下打印二叉树

    public ArrayList<Integer> printTree(TreeNode root){
        ArrayList<Integer> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if (root == null){
            return res;
        }
        queue.add(root);
        while (!queue.isEmpty()){
            if (queue.peek().left != null){
                queue.add(queue.peek().left);
            }
            if (queue.peek().right != null){
                queue.add(queue.peek().right);
            }
            res.add(queue.poll().val);
        }
        return res;
    }

题目二：分行从上到下打印二叉树

    public ArrayList<ArrayList<Integer>> printTree(TreeNode root){
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if (root == null){
            return res;
        }

        queue.add(root);
        int curLine = 1;
        int nextLine = 0;
        ArrayList<Integer> list = new ArrayList<>();
        while (!queue.isEmpty()){
            if (queue.peek().left != null){
                queue.add(queue.peek().left);
                nextLine++;
            }
            if (queue.peek().right != null){
                queue.add(queue.peek().right);
                nextLine++;
            }
            list.add(queue.poll().val);
            curLine--;
            if (curLine == 0){
                curLine = nextLine;
                nextLine = 0;
                res.add(new ArrayList<>(list));
                list.clear();
            }
        }
        return res;
    }

题目三：之字形打印二叉树

    public ArrayList<ArrayList<Integer>> printTree(TreeNode root){
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        Stack<TreeNode> stack0 = new Stack<>();
        Stack<TreeNode> stack1 = new Stack<>();
        if (root == null){
            return res;
        }
        int dir = 1; //0代表从左向右打印,1代表从右向左打印
        int curLine = 1;
        int nextLine = 0;
        stack0.push(root);
        ArrayList<Integer> list = new ArrayList<>();
        while (!stack0.isEmpty() || !stack1.isEmpty()){
            if (dir == 1){
                if (stack0.peek().left != null){
                    stack1.push(stack0.peek().left);
                    nextLine++;
                }
                if (stack0.peek().right != null){
                    stack1.push(stack0.peek().right);
                    nextLine++;
                }
            }else {
                if (stack1.peek().right != null){
                    stack0.push(stack1.peek().right);
                    nextLine++;
                }
                if (stack1.peek().left != null){
                    stack0.push(stack1.peek().left);
                    nextLine++;
                }
            }

            if (dir == 1){
                list.add(stack0.pop().val);
            }else {
                list.add(stack1.pop().val);
            }

            curLine--;
            if (curLine == 0){
                curLine = nextLine;
                nextLine = 0;
                dir = (dir == 0 ? 1 : 0);
                res.add(new ArrayList<>(list));
                list.clear();
            }
        }
        return res;
    }