给定一个二叉树的 根节点 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
代码
def rightSideView(root):
# 右视图是最右边的节点,用dfs先从右子树遍历
def dfs(node,level):
if not node:
return
if len(ans) == level:
# len(ans)代表该收集第len(ans)层的答案了,每层最先遍历的是右节点,直接收集
ans.append(node.val)
dfs(node.right,level + 1)
dfs(node.left,level + 1)
ans = []
dfs(root,0)
return ans