【python/M/142】Linked List Cycle II

最新推荐文章于 2022-12-06 22:57:02 发布

原创最新推荐文章于 2022-12-06 22:57:02 发布 · 294 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #python

收录于

LeetCode 专栏收录该内容

79 篇文章

订阅专栏

本文介绍了一种解决特定二叉树问题的方法：通过层次遍历来获取每层的最后一个元素。使用递归方式实现层次遍历，代码清晰易懂，适用于理解和实践二叉树的相关操作。

题目

这里写图片描述

基本思路

这道题要求返回的是每一层的最后一个元素，既然提到了层，想必就是层次遍历了，采用层序遍历递归写法。
层次遍历代码需要抽象出来，牢牢记住了。

代码

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

class Solution(object):
    def rightSideView(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        res = []
        self.levelOrder(root,0,res)
        return [level[-1] for level in res]

    def levelOrder(self,root,level,res):
        if not root:
            return
        if level == len(res):
            res.append([])
        res[level].append(root.val)
        self.levelOrder(root.left,level+1,res)
        self.levelOrder(root.right,level+1,res)