从上往下打印二叉树-剑指Offer

Posted 2020-07-31 RosenDing

从上往下打印二叉树

题目描述

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

思路

1. 数据结构：队列
2. 将树按层遍历，先将根节点的左右子节点先后进入队列，然后再先后从队列取出，然后将取出的节点的左右子节点再入队列
3. 循环进行，直到队列为空

代码

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/**
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) {
	        Queue<TreeNode> queue = new LinkedList<TreeNode>(); 
	        queue.offer(root);
	        while (!queue.isEmpty()) {
	        	root = queue.poll();
	        	list.add(root.val);
	        	if (root.left != null) {
	        		queue.offer(root.left);
	        	}
	        	if (root.right != null) {
	        		queue.offer(root.right);
	        	}
	        }
        }
        return list;
    }
}