leetcode -- Binary Tree Right Side View - 중점

1552 단어 LeetCode
https://leetcode.com/problems/binary-tree-right-side-view/
처음에 제 생각은 루트에서 오른쪽 나무로 쭉 가다가 오른쪽 나무가 없을 때까지 가다가 왼쪽 나무로 계속 돌아가는 거예요...그러나 이렇게 해서 사실 한 가지 상황을 소홀히 한 것은 바로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]

좋은 웹페이지 즐겨찾기