【数据结构与算法 | 灵神题单 | 二叉搜索树篇】力扣653
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 题解:
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);
}
}