《剑指offer》—— 22. 从上往下打印二叉树（Java）

推荐

完整《剑指Offer》算法题解析系列请点击 👉 《剑指Offer》全解析 Java 版

 

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        
    }
}

 

思路：

就是使用队列来进行层次遍历；

首先了解一下队列：

队列是一种特殊的线性表，它只允许在表的前端进行删除操作，而在表的后端进行插入操作。

LinkedList类实现了Queue接口，因此我们可以把LinkedList当成Queue来用。

队列的几个常用方法：

1. poll() ，返回队列中的第一个元素，并且在队列中删除这个元素。
2. peek()，返回队列中的第一元素，队列为空时，返回null。
3. element() ，返回队列中的第一元素，队列为空时，抛出一个异常。
4. add()，向队列中添加一个元素，失败时会抛出异常。（不推荐）
5. offer()，向队列中添加一个元素，失败时会返回false。（推荐）

 

具体思路可看代码中注释。

 

实现：

import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        if (root == null){
            return list;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        // 1.将树的的根结点放入队列
        queue.offer(root);

        // 2.队列不为空时执行循环3、4、5、6
        while(!queue.isEmpty()){
            // 3.将队列中的第一个元素作为树的当前结点，并在队列中删除这个元素；
            TreeNode t = queue.poll();
            // 4. 将树的当前节点的 左节点放入队列
            if (t.left != null){
                queue.offer(t.left);
            }
            // 5. 将树的当前节点的 右节点放入队列
            if (t.right != null){
                queue.offer(t.right);
            }
            // 6.将当前节点存入list集合中
            list.add(t.val);
    }
     return list;
    }
}

 

看完之后，如果还有什么不懂的，可以在评论区留言，会及时回答更新。

点个关注，不再迷路

这里是猿兄，为你分享程序员的世界。

非常感谢各位大佬们能看到这里，如果觉得文章还不错的话， 求点赞👍 求关注💗 求分享👬求评论📝 这些对猿兄来说真的 非常有用！！！

注： 如果猿兄这篇博客有任何错误和建议，欢迎大家留言，不胜感激！