LeetCode653. 两数之和 IV - 输入 BST

最新推荐文章于 2024-03-31 12:09:16 发布

原创最新推荐文章于 2024-03-31 12:09:16 发布 · 123 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

41 篇文章

订阅专栏

本文详细解析了LeetCode653题“两数之和IV-输入BST”的解决方案，该题要求在二叉搜索树中寻找两个元素，使它们的和等于给定的目标值。文章提供了一个使用广度优先搜索和集合数据结构实现的高效算法。

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

LeetCode653. 两数之和 IV - 输入 BST

给定一个二叉搜索树和一个目标结果，如果 BST 中存在两个元素且它们的和等于给定的目标结果，则返回 true。

案例 1:

输入: 
    5
   / \
  3   6
 / \   \
2   4   7

Target = 9

输出: True

案例 2:

输入: 
    5
   / \
  3   6
 / \   \
2   4   7

Target = 28

输出: False

class Solution:    
    def findTarget(self, root, k):
        """
        :type root: TreeNode
        :type k: int
        :rtype: bool
        """
        if not root:
            return False
        nums = set()
        queue = [root]
        while queue:
            temp = queue.pop(0)
            nums.add(temp.val)
            if temp.left:
                queue.append(temp.left)
            if temp.right:
                queue.append(temp.right)
        for i in nums:
            if k-i in nums and i!=k/2:
                return True
        return False