##題目描述 從上往下列印出二叉樹的每個節點,同層節點從左至右列印,
思路
層序遍歷,
時間復雜度O(n),空間復雜度O(1),
代碼
import java.util.Queue;
import java.util.LinkedList;
import java.util.ArrayList;
/**
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) return list;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
while(queue.peek() != null) {
TreeNode tmp = queue.poll();
list.add(tmp.val);
if(tmp.left != null) {
queue.offer(tmp.left);
}
if(tmp.right != null) {
queue.offer(tmp.right);
}
}
return list;
}
}
筆記
Java中null與[]不相等,回傳值需要按照型別回傳,不能直接回傳null,
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/85162.html
標籤:其他
上一篇:堆疊的壓入、彈出序列
下一篇:二叉搜索樹的后序遍歷序列
