題目描述:
從上往下打印出二叉樹(shù)的每個(gè)節(jié)點(diǎn),同層節(jié)點(diǎn)從左至右打印。
分析:
二叉樹(shù)的遍歷:
前序遍歷:根->左->右
中序遍歷:左->根->右
后序遍歷:左->右->根
層次遍歷:一層一層,從左向右遍歷。
在層次遍歷中,元素需要儲(chǔ)存有先進(jìn)先出的特性,所以選用隊(duì)列存儲(chǔ)。
【思路:】
用隊(duì)列來(lái)存儲(chǔ)上一層的節(jié)點(diǎn),用作遍歷下一層的跟節(jié)點(diǎn)集合。
- 將根結(jié)點(diǎn)加入隊(duì)列。
- 循環(huán)出隊(duì),并打印當(dāng)前元素,若該結(jié)點(diǎn)有左子樹(shù),則將其加入隊(duì)列,若有右子樹(shù),將其加入隊(duì)列。
- 直到隊(duì)列為空,表明已經(jīng)打印完所有結(jié)點(diǎn)。
代碼:
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> list = new ArrayList<Integer>();
if(root == null)return list;
Queue<TreeNode> q = new LinkedList<TreeNode>();
q.add(root);
while(!q.isEmpty()){
TreeNode tmp = q.remove();
list.add(tmp.val);
if(tmp.left!=null) q.add(tmp.left);
if(tmp.right!=null) q.add(tmp.right);
}
return list;
}
}

注意:當(dāng)根節(jié)點(diǎn)為空的時(shí)候,應(yīng)該返回的是新new出來(lái)的ArrayList,而不是返回null,否則會(huì)報(bào)錯(cuò)。
小結(jié):
關(guān)于隊(duì)列Queue的基本方法:
add() 增加一個(gè)元索 如果隊(duì)列已滿,則拋出一個(gè)IIIegaISlabEepeplian異常
remove() 移除并返回隊(duì)列頭部的元素 如果隊(duì)列為空,則拋出一個(gè)NoSuchElementException異常
element() 返回隊(duì)列頭部的元素 如果隊(duì)列為空,則拋出一個(gè)NoSuchElementException異常
offer() 添加一個(gè)元素并返回true 如果隊(duì)列已滿,則返回false
poll() 移除并返問(wèn)隊(duì)列頭部的元素 如果隊(duì)列為空,則返回null
peek() 返回隊(duì)列頭部的元素 如果隊(duì)列為空,則返回null
put() 添加一個(gè)元素 如果隊(duì)列滿,則阻塞
take() 移除并返回隊(duì)列頭部的元素 如果隊(duì)列為空,則阻塞