（Java）leetcode-653 Two Sum IV - Input is a BST (两数之和 IV - 输入 BST)

最新推荐文章于 2024-07-22 14:15:06 发布

Mr.Bean-Pig

最新推荐文章于 2024-07-22 14:15:06 发布

阅读量138

点赞数

分类专栏：算法题解文章标签： java 算法 leetcode hashmap hashtable

本文链接：https://blog.csdn.net/z714405489/article/details/106602578

版权

算法题解专栏收录该内容

140 篇文章 0 订阅

订阅专栏

题目描述

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

思路

参考leetcode-1 Two Sum 。经典的第一题中，由于需要记录符合的数据位置，所以使用hashMap来记录已经遍历过的数据。而此题中无须记录位置，因此使用hashSet即可。

代码

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