題目描述
從上往下列印出二叉樹的每個節點,同層節點從左至右列印。
import java.util.ArrayList;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
思路:層次周遊,用arraylist模拟一個隊列來存儲相應的TreeNode
*/
import java.util.LinkedList;
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
//注意不能直接傳回null,不然樹為空時,不能通過
if(root ==null) return new ArrayList<Integer>();
//1、定義一個Arraylist對象來存儲二叉樹的節點值
ArrayList<Integer> arr = new ArrayList<>();
//2、用linkedList來做層次周遊
LinkedList<TreeNode> list = new LinkedList<>();
list.add(root);
while(!list.isEmpty()){
TreeNode temp = list.removeFirst();
if(temp.left!=null){
list.add(temp.left);
}
if(temp.right!=null){
list.add(temp.right);
}
arr.add(temp.val);
}
return arr;
}
}