Leetcode199 二叉樹的右視圖 - JAVA

題目傳送門

題目

給定一棵二叉樹，想象自己站在它的右側，按照從頂部到底部的順序，返回從右側所能看到的節點值。

示例:

輸入: [1,2,3,null,5,null,4]
輸出: [1, 3, 4]
解釋:

   1            <---
 /   \
2     3         <---
 \     \
  5     4       <---

解法

BFS 解法

思路

逐層遍歷，把每一層的最後一個元素拿出來即可。
套路模板，僞代碼如下：

Queue queue = new Queue(); // 準備一個隊列，我喜歡用Java裏的LinkedList,隨心所欲，想從哪邊出就從哪邊出
if(root != null) {
	queue.add(root); // 把根結點加進來
}
while(!queue.isEmpty()) { // 結束條件就是遍歷完最後一層，爲空退出
 	for(queue) {   // 特別注意，要用成計數的方式來遍歷，因爲要讓上一層出隊，並且讓上一層的子節點入隊，別一邊出一邊進，仔細想想就知道問題在哪了。
 		node = queue.poll();
 		if(null != node.left )
 			queue.add(node.left);
 		if(null != node.right)
 			queue.add(node.right);
 	}
}

代碼

class Solution {
    List<Integer> res = new ArrayList<>();
    public List<Integer> rightSideView(TreeNode root) {
        if(root == null) return res;
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()) {
            res.add(queue.getLast().val);
            for(int i=queue.size();i>0;i--) {
                TreeNode poll = queue.removeFirst();
                if(null != poll.left) {
                    queue.add(poll.left);
                }
                 if(null != poll.right) {
                    queue.add(poll.right);
                }
            }
        }
        return res;
    }
}