LeetCode 653. 两数之和 IV - 输入二叉搜索树

653. 两数之和 IV - 输入二叉搜索树

难度：$easy$

---

题目描述

给定一个二叉搜索树 $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,10^4]$.
• $-10^4<=Node.val<=10^4$
• 题目数据保证，输入的 $root$ 是一棵 有效 的二叉搜索树
• $-10^5<=k<=10^5$

---

算法

(深度优先搜索 + 哈希表)

我们可以使用深度优先搜索的方式遍历整棵树，用哈希表记录遍历过的节点的值。

对于一个值为 x 的节点，我们检查哈希表中是否存在 k−x 即可。如果存在对应的元素，那么我们就可以在该树上找到两个节点的和为 k；否则，我们将 x 放入到哈希表中。

如果遍历完整棵树都不存在对应的元素，那么该树上不存在两个和为 k 的节点。

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 为二叉搜索树的大小。我们需要遍历整棵树一次。

• 空间复杂度 : $O(n)$，其中 $n$ 为二叉搜索树的大小。

C++ 代码

/*** Definition for a binary tree node.* struct TreeNode {*     int val;*     TreeNode *left;*     TreeNode *right;*     TreeNode() : val(0), left(nullptr), right(nullptr) {}*     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}*     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:unordered_set<int> hash;bool findTarget(TreeNode* root, int k) {if (!root) return false;if (hash.count(k - root->val)) return true;hash.insert(root->val);return findTarget(root->left, k) || findTarget(root->right, k);}
};

---