865. 具有所有最深结点的最小子树

本文介绍了两种寻找具有最深深度子树的方法:一种是一次深度优先搜索(DFS),另一种是两次DFS,先找到最深节点再确定答案节点。通过递归方式实现,适用于二叉树结构。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

两个思路吧,第二个略麻烦。

第一个是一次dfs

class Solution:
    def dfs(self, root):
        if not root:
            return (root, 0)
        left_dict = self.dfs(root.left)
        right_dict = self.dfs(root.right)

        if left_dict[1] == right_dict[1]:
            return (root, left_dict[1]+1)
        elif left_dict[1] > right_dict[1]:
            return (left_dict[0], left_dict[1]+1)
        else:
            return (right_dict[0], right_dict[1]+1)


    def subtreeWithAllDeepest(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        ans_dict = self.dfs(root)
        return ans_dict[0]

第二个思路是两次dfs,一次找最深节点,第二次找到答案节点。

class Solution:
    def __init__(self):
        self.deepest = 0
        self.deep_node = []
        self.ans_node = None

    def dfs(self, root, deep_now):
        if not root:
            return
        if not (root.right or root.left):
            if deep_now > self.deepest:
                self.deepest = deep_now
                self.deep_node = []
                self.deep_node.append(root)
                return
            if deep_now == self.deepest:
                self.deep_node.append(root)
                return

            self.dfs(root.left, deep_now + 1)
            self.dfs(root.rignt, deep_now + 1)
        return

    def dfs_1(self, root):
        if root in self.deep_node:
            return 1
        k1 = 0
        k2 = 0
        if not root.left:
            k1 = self.dfs_1(root.left)
        if not root.right:
            k2 = self.dfs_1(root.right)
        if k1 + k2 == len(self.deep_node) and self.ans_node:
            self.ans_node = root

        return k1 + k2

    def subtreeWithAllDeepest(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        self.dfs(root,0)
        self.dfs_1(root)

        return self.ans_node

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值