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]