【今日刷题】LeetCode 199.二叉树的右视图（中等）

今日刷题：LeetCode 199.二叉树的右视图（中等）
题目描述：
给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

示例 1:

输入: [1,2,3,null,5,null,4]
输出: [1,3,4]
示例 2:

输入: [1,null,3]
输出: [1,3]
示例 3:

输入: []
输出: []

提示:

二叉树的节点个数的范围是 [0,100]

-100 <= Node.val <= 100

这道题目需要拿到二叉树的 右视图 ，也就是右边的第一个节点，不过要注意右边的第一个节点不一定是右节点，也有可能右节点为空，那么就是左节点了，如下：

那么解题思路的话，有两种 dfs 或者 bfs

DFS 解决

先说 dfs 解决的思路，可以按照 根节点 -> 右节点 -> 左节点 的顺序进行遍历

由于每一层我们只需要加最右边的节点，但是遍历的话，会遍历这一层的多个节点

因此通过一个 depth 变量来记录遍历的深度，由于每一层都会添加一个节点，这里假设节点放在 res 数组中了，那么如果 depth ==res ，说明上一层刚遍历完，现在遍历的是新的一层的第一个节点，我们定义的递归顺序就是先便利右节点，因此一定会先遍历 右视图 的节点，将该节点加入 res 中即可

代码如下 ：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    List<Integer> res = new ArrayList<>();

    public List<Integer> rightSideView(TreeNode root) {
        dfs(root,0);
        return res;
    }

    public void dfs(TreeNode root,int high){
        if(root==null){
            return;
        }
        if(high==res.size()){
            res.add(root.val);
        }
        dfs(root.right,high+1);
        dfs(root.left,high+1);
    }
}

BFS 解决

BFS 来解决的话，可以将每一层的节点加入到队列中，先加入左节点，再加入右节点，那么队列中的最后一个元素肯定就是 右视图 了，加入到结果集中即可

代码如下 ：

// 层序遍历，将每一层的最后一个节点加入结果集合
public class Solution {
    /**
     * 解法：队列，迭代。
     * 每次返回每层的最后一个字段即可。
     */
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> que = new LinkedList<>();

        if (root == null) {
            return list;
        }

        que.offerLast(root);
        while (!que.isEmpty()) {
            int levelSize = que.size();

            for (int i = 0; i < levelSize; i++) {
                TreeNode poll = que.pollFirst();

                if (poll.left != null) {
                    que.addLast(poll.left);
                }
                if (poll.right != null) {
                    que.addLast(poll.right);
                }

                if (i == levelSize - 1) {
                    list.add(poll.val);
                }
            }
        }

        return list;
    }
}