LeetCode199 Binary Tree Right Side View

LeetCode199 Binary Tree Right Side View

問題來源LeetCode199

問題描述

Given a binary tree, imagine yourself standing on the right side of it, return the values of the nodes you can see ordered from top to bottom.

For example:

Given the following binary tree,
   1            <---
 /   \
2     3         <---
 \     \
  5     4       <---
You should return [1, 3, 4].

這道題就是假設你站在二叉樹的右邊，返回能夠看到的節點的值。

問題分析

這道題其實就是對樹進行層遍歷，然後保存每層最後一個元素的值。層遍歷實現的方法很多，這裏我用的是隊列。最後出列的元素就是該層的最後一個元素。

代碼如下

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public List<Integer> rightSideView(TreeNode root) {
    List<Integer> res = new ArrayList<>();
    if(root==null) return res;
    Queue<TreeNode> queue = new ArrayDeque<>();
    Queue<TreeNode> queue2 = new ArrayDeque<>();
    queue.add(root);
    while (!queue.isEmpty()){
        TreeNode node  =queue.poll();
        if(node.left!=null) queue2.add(node.left);
        if(node.right!=null) queue2.add(node.right);
        if(queue.isEmpty()){
            res.add(node.val);
            queue=queue2;
            queue2=new ArrayDeque<>();
        }
    }
    return res;
}

LeetCode學習筆記持續更新

GitHub地址 https://github.com/yanqinghe/leetcode

CSDN博客地址 http://blog.csdn.net/yanqinghe123/article/category/7176678