653. 两数之和 IV - 输入 BST 难度[简单]
给定一个二叉搜索树 root 和一个目标结果 k,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
示例 1:
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-scVEmxL3-1647838175293)(https://s3-us-west-2.amazonaws.com/secure.notion-static.com/fb55a0bd-76e7-461e-a63d-3dca62085624/Untitled.png)]
输入: root = [5,3,6,2,4,null,7], k = 9
输出: true
示例 2:
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ApqERvTi-1647838175295)(https://s3-us-west-2.amazonaws.com/secure.notion-static.com/890c18e3-89ef-4bdc-8c52-3f0314d3ca24/Untitled.png)]
输入: 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法一:hashset + 遍历求和
class Solution {
Set<Integer> set = new HashSet<>();
Boolean flag = false;
public boolean findTarget(TreeNode root, int k) {
dfs(root, k);
return flag;
}
void dfs (TreeNode root, int k) {
if (root == null) return;
if (flag == false) {
if (set.contains(k-root.val)) flag = true;
} else {
return ;
}
set.add(root.val);
if (root.left != null) dfs(root.left, k);
if (root.right != null) dfs(root.right, k);
}
}
此文章创于本人学习时的记录,如有错误或更优解还请指出