【数据结构与算法 | 灵神题单 | 二叉搜索树篇】力扣653

Mephisto.java2024-09-22 15:06

1. 力扣653：两数之和IV - 输入二叉搜索树

1.1 题目：

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

示例 1：

复制代码

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

示例 2：

复制代码

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

提示:

二叉树的节点个数的范围是 [1, 104].
-104 <= Node.val <= 104
题目数据保证，输入的 root 是一棵有效的二叉搜索树
-105 <= k <= 105

1.2 思路：

用哈希表存储遍历过的元素，如果在哈希表中找到了自己的另一半，返回true，否则继续查找。

1.3 题解：

java 复制代码

class Solution {
    // 使用哈希表
    HashSet<Integer> hashset = new HashSet<>();
    boolean flag = false;
    public boolean findTarget(TreeNode root, int k) {
        dfs(root, k);
        return flag;
    }
    //dfs的遍历顺序其实不重要
    private void dfs(TreeNode node, int k) {
        if(node == null) return;

        dfs(node.left, k);
        // 如果在哈希表中没找到另一个加起来等于k的元素
        // 就把该节点值加入到哈希表中
        // 继续判断后续节点的值
        
        // 如果找到了，由题意，返回true
        if(!hashset.contains(k - node.val)){
            hashset.add(node.val);
        }else{
            flag = true;
            return;
        }

        dfs(node.right, k);
    }
}