199. 二叉树的右视图
题目
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <---
/ \
2 3 <---
\ \
5 4 <---
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-right-side-view
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解思路
1>源代码
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def rightSideView(self, root: TreeNode) -> List[int]:
depth = 0
ans = []
hashset = set()
def helper(root, depth):
if depth not in hashset:
hashset.add(depth)
ans.append(root.val)
if root.right != None:
helper(root.right, depth+1)
if root.left != None:
helper(root.left, depth+1)
if root!=None:
helper(root, depth)
return ans
2>算法介绍
本题难度不大,对于一棵树的"右视图",我们很容易想到使用dfs算法。常规的dfs算法一般是先遍历左子树再遍历右子树,这里我们只需要修改一下先后顺序就可以达到目的。当然,由于我们想得到的是一层中的最右边节点,所以我们可以维护一个hash集合用来存储已经访问过的层数,如果当前访问的节点所在层数已经被访问过,则说明最右节点已经找到了,不将当前节点的值加入答案中。