剑指offer32题：从上到下打印二叉树

最新推荐文章于 2024-07-14 14:13:59 发布

原创最新推荐文章于 2024-07-14 14:13:59 发布 · 160 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#队列 #二叉树 #数据结构 #java #算法

java 同时被 2 个专栏收录

28 篇文章

订阅专栏

剑指offer刷题

27 篇文章

订阅专栏

本文介绍了一种二叉树的层次遍历算法，通过队列实现从上到下、从左至右的节点访问。代码示例清晰展示了如何利用队列特性，逐层遍历并输出二叉树各层节点。

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

先上代码，然后在分析。

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 {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer> > result = new ArrayList<ArrayList<Integer> >();
        if(pRoot == null){
            return result;
        }
        ArrayList<Integer> temp = new ArrayList<Integer>();
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(pRoot);
        int start = 0, end = 1;
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();
            temp.add(node.val);
            start ++;
            if(node.left != null){
                queue.add(node.left);
            }
            if(node.right != null){
                queue.add(node.right);
            }
            if(start == end){
                end = queue.size();
                start = 0;
                result.add(temp);
                temp = new ArrayList<Integer>();
            }
        }
        return result;
    }
}

分析如下：
利用队列先进先出的特点，先把根节点放入队列，然后弹出到list中，在依次访问左右节点，如果不为空，则依次放入队列。因为需要把每一层单独放在一个list中，所以需要使用end = queue.size();记录每层节点的个数。每次弹出一个就执行start ++;，当start == end就代表这一层遍历结束，就把保存的list存入最终结果中，并且start = 0，重新开始下一层的遍历。