一刷211-剑指 Offer II 056. 二叉搜索树中两个节点之和（同：653两数之和IV-输入BST）（e）

丿唐宋

已于 2022-03-28 21:06:15 修改

阅读量170

点赞数

于 2022-03-19 10:45:26 首次发布

本文链接：https://blog.csdn.net/m0_45170946/article/details/123590160

版权

二叉搜索树目标和节点查找递归哈希集合

关键词由CSDN通过智能技术生成

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

在这里插入图片描述

输入: root = [5,3,6,2,4,null,7], k = 9
输出: true

在这里插入图片描述

输入: root = [5,3,6,2,4,null,7], k = 28
输出: false
 
提示:
二叉树的节点个数的范围是  [1, 104].
-104 <= Node.val <= 104
root 为二叉搜索树
-105 <= k <= 105
------------------------
思路：
在递归搜索过程中记录下相应的节点值（使用 Set 集合），
如果在遍历某个节点 x 时发现集合中存在 k−x.val，
说明存在两个节点之和等于 k，返回 True，若搜索完整棵树都没有则返回 False。
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {//哈希查找不需要返回下标，则用set
    Set<Integer> set = new HashSet<>();
    public boolean findTarget(TreeNode root, int k) {
        if (root == null) return false;
        if (set.contains(k - root.val)) return true;
        set.add(root.val);
        return findTarget(root.left, k) || findTarget(root.right, k);//递归查找
    }
}