题目

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

思路

  • 按照二叉树层次遍历的过程修改程序
  • 首先判断根节点是否存在，然后遍历根节点，并判断根节点的左右两个子节点是否存在
  • 如果左右或者有一个存在，那么就可以执行下一次的循环，直到发现不再存在根节点

代码

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> templist = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if(root==null)
            return templist;
        queue.add(root);
        while(queue.size()!=0){
            int len = queue.size();
            
            for(int i=0;i<len;i++){
                TreeNode temp =queue.poll();
                templist.add(temp.val);
                
                if(temp.left!=null){
                    queue.add(temp.left);
                }
                if(temp.right!=null){
                    queue.add(temp.right);
                }
            }
        }
     
        return templist;
    }
}