Leetcode 199.二叉树的右视图

Time: 20190903
Type: Medium

题目描述

给定一棵二叉树，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

示例:

输入: [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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

这是一类题型，既然是树，那么就可以有上下左右四种不同的视图，需要能做到举一反三。

右视图或者左视图这种横向的，可以直接求出层序遍历序列，然后取最右边的数值即可。

代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def rightSideView(self, root: TreeNode) -> List[int]:
        res = [] # 用于存储层序遍历的结果
        if not root:
            return res
        
        def dfs(root, level):
            if not root: # 递归出口很重要
                return 
            if len(res) == level:
                res.append([])
                
            res[level] += [root.val]
            
            dfs(root.left, level + 1)
            dfs(root.right, level + 1)
        dfs(root, 0)        
        ret = [num[-1] for num in res]
        return ret

或者层序遍历时只保存一个结果，即不断替换，深度优先遍历也是一步步读取数据的。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def rightSideView(self, root: TreeNode) -> List[int]:
        res = {
     } # 用于存储层序遍历的结果
        if not root:
            return res

        def dfs(root, level):
            if not root:
                return 
            # 只保存一层的最后一个结点    
            res[level] = root.val   
            dfs(root.left, level + 1)
            dfs(root.right, level + 1)
            
        dfs(root, 0)
        # print(list(res.values()))
        return list(res.values())

骨架是一样的。

END.