今日刷题: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;}
}