剑指offer（把二叉树打印成多行）

最新推荐文章于 2019-08-12 16:51:12 发布

失败的墨鱼

最新推荐文章于 2019-08-12 16:51:12 发布

阅读量125

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer

本文链接：https://blog.youkuaiyun.com/weixin_41999706/article/details/89284245

剑指offer 专栏收录该内容

31 篇文章

订阅专栏

本文详细介绍了二叉树的层次遍历算法，包括从上到下按层打印二叉树的方法，以及如何实现每一层结点从左至右输出。通过使用队列和栈的数据结构，实现了二叉树的层次遍历，并提供了完整的Java代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

把二叉树打印成多行

题目描述
从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;

public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        ArrayList<Integer> tmp = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if(pRoot == null) return res;
        queue.offer(pRoot);
        TreeNode last = pRoot;
        TreeNode nlast = null;
        while( ! queue.isEmpty()){
            pRoot = queue.poll();
            tmp.add(pRoot.val);
            if(pRoot.left != null){
                queue.offer(pRoot.left);
                nlast = pRoot.left;
            }
            if(pRoot.right != null){
                queue.offer(pRoot.right);
                nlast = pRoot.right;
            }
            if(pRoot == last && !queue.isEmpty() /* 这一个原因是上边有一个 poll*/){
                res.add(new ArrayList<>(tmp));
                last= nlast;
                tmp.clear();
            }
        }
        res.add(new ArrayList<>(tmp)); //加入的是最后一行
        return res;
    
    }
    
}

last是当前最后一个，nlast是下一个最后一个

所以当pRoot到达last的时候，把last变成nlast nlast是最后一行的最右边

leetcode 107 二叉树的层次遍历 II

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
     public List<List<Integer>> levelOrderBottom(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if(root == null) return res;
        
        Queue<TreeNode> queue = new LinkedList<>();
        Stack<List<Integer>> stack = new Stack<>();
        queue.offer(root);
        
        while(! queue.isEmpty()){
            int size = queue.size();
            List<Integer> list = new ArrayList<>();
            for(int i = 0; i < size; i++){
                TreeNode cur = queue.poll();
                if(cur.left != null) queue.offer(cur.left);
                if(cur.right != null) queue.offer(cur.right);
                list.add(cur.val);
            }
            
            stack.add(list);
        }
        
        while(! stack.isEmpty()){
            res.add(stack.pop());
        }
        
        return res;
    }
}