【leetcode系列】【算法】【简单】两数之和 IV - 输入 BST（Binary Search Tree）

最新推荐文章于 2024-06-21 17:44:59 发布

冰临天下

最新推荐文章于 2024-06-21 17:44:59 发布

阅读量118

点赞数

分类专栏： leetcode系列

本文链接：https://blog.youkuaiyun.com/songyuwen0808/article/details/105675827

版权

leetcode系列专栏收录该内容

139 篇文章

订阅专栏

本文介绍了解决LeetCode上BST中两数之和IV问题的算法思路，通过使用哈希表和树的遍历方式，高效地找到目标和的两个数。详细解释了如何利用哈希表保存已遍历的数字，并在遍历过程中检查是否存在目标和的另一数。

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

题目：

题目链接：https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst/

解题思路：

类似两数之和的解题思路

添加一个hash表保存已经遍历过的数字，只是需要将遍历方式修改为树的遍历方式

代码实现：

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

class Solution:
    def findTarget(self, root: TreeNode, k: int) -> bool:
        if not root:
            return False

        rec = set()
        stack = [root]
        while stack:
            node = stack[-1]
            stack.pop()
            other_num = k - node.val
            if other_num in rec:
                return True

            rec.add(node.val)
            if node.left:
                stack.append(node.left)

            if node.right:
                stack.append(node.right)

        return False