算法—层序遍历二叉树

题目

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

分析

此题即是层序遍历二叉树，引入队列，先将根节点入队，接着循环执行出队操作，每出队一个节点，就将其放入列表，同时将该节点的左孩子和右孩子入队。直到队列为空为止。

代码

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) {
        //层序遍历，引入队列(这里的队列用ArrayList实现)
        ArrayList<Integer> list = new ArrayList<>();
        if(root == null){
            return list;
        }
        ArrayList<TreeNode> queue = new ArrayList<>();
        queue.add(root); //先把根节点放入队列
        while(!queue.isEmpty()){
            TreeNode tree = queue.remove(0);  //出队
            list.add(tree.val);
            if(tree.left != null){
                queue.add(tree.left);  //左孩子入队
            }
            if(tree.right != null){
                queue.add(tree.right);  //右孩子入队
            }
        }
        return list;
    }
}

运行结果