leetcode 653. Two Sum IV - Input is a BST（两个数之和，输入是BST）

蓝羽飞鸟

于 2020-12-15 20:42:19 发布

阅读量131

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode 算法

本文链接：https://blog.youkuaiyun.com/level_code/article/details/111239821

leetcode 专栏收录该内容

490 篇文章

订阅专栏

这篇博客讨论了如何在一个给定的二叉搜索树（BST）中判断是否存在两个节点，它们的值之和等于给定的目标数k。通过前序遍历和使用HashSet来存储已遍历过的节点值，可以有效地解决这个问题。当遍历到一个节点时，检查HashSet中是否存在k减去当前节点值的元素，如果存在则返回true，否则将当前节点值添加到HashSet中并继续遍历。这种方法确保了在BST的性质下有效查找目标和。

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

Given the root of a Binary Search Tree and a target number k, return true if there exist two elements in the BST such that their sum is equal to the given target.

Example 1:
在这里插入图片描述
Constraints:

The number of nodes in the tree is in the range [1, 104].
-104 <= Node.val <= 104
root is guaranteed to be a valid binary search tree.
-105 <= k <= 105

给出BST，问其中是否存在两个节点值的和=k

思路：
涉及二叉数的遍历，Two sum。
二叉树的遍历随便选一个，这里选前序遍历，Two sum就用一个HashSet保存已经遍历过的数字，如果后面set里面保存的有k-root.val，就说明存在。

    public boolean findTarget(TreeNode root, int k) {
        HashSet<Integer> set = new HashSet<>();
        
        return preTraversal(root, set, k);
        
    }
    
    boolean preTraversal(TreeNode root, HashSet<Integer> set, int k) {
        if(root == null) return false;
        if(set.contains(k - root.val)) return true;
        set.add(root.val);
        if(preTraversal(root.left, set, k)) return true;
        if(preTraversal(root.right, set, k)) return true;
        return false;
    }