題目描述
從上往下列印出二叉樹的每個節點,同層節點從左至右列印。
就是二叉樹的層次周遊,用隊列,根節點入隊列,每次列印一個節點時,如果該節點有子節點,則子節點放入隊尾,直到隊列中所有的節點都被列印出來。
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/**
* @author yuan
* @date 2019/2/14
* @description 層次周遊
*/
public class 從上往下列印二叉樹 {
static class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> result = new ArrayList<>();
if (root == null) {
return result;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
result.add(node.val);
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
return result;
}
}