問題描述:
從上往下打印出二叉樹的每個節點,同層節點從左至右打印。
思路:
按照層次遍歷的方法,使用隊列輔助。
1.將根結點加入隊列。
2.循環出隊,打印當前元素,若該結點有左子樹,則將其加入隊列,若有右子樹,將其加入隊列。
3.直到隊列爲空,表明已經打印完所有結點。
代碼:
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> result = new ArrayList<Integer>();
if(root == null){
return result;
}
Queue<TreeNode> q = new LinkedList<TreeNode>();
q.add(root);
while(!q.isEmpty()){
TreeNode tmp = q.remove();
result.add(tmp.val);
if(tmp.left != null){
q.add(tmp.left);
}
if(tmp.right != null){
q.add(tmp.right);
}
}
return result;
}
}