https://leetcode.com/problems/binary-tree-right-side-view/
一开始我的思路是从root一直递归向右子树走,一直走到没有右子树,然后再走到左子树,继续递归。。。。但这样其实忽略了一种情况就是。e.g.[1,2,3,4], 如果这种递归的话,4就搜索不到。
这里其实就是level order tranversal。取level order最后的一个元素就行了
class Solution(object):
def dfs(self, level, root, res):
if root:
if len(res) < level + 1:
res.append([])
res[level].append(root.val)
self.dfs(level + 1, root.left, res)
self.dfs(level + 1, root.right, res)
def rightSideView(self, root):
"""
:type root: TreeNode
:rtype: List[int]
"""
res = []
self.dfs(0, root, res)
return [x[-1] for x in res]