【LeetCode】第653题——两数之和 IV - 输入 BST(难度:简单)
题目描述
给定一个二叉搜索树和一个目标结果,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
-
示例 1:
输入:5 / \ 3 6 / \ \ 2 4 7
Target = 9
输出: True -
示例 2:
输入:5 / \ 3 6 / \ \ 2 4 7
Target = 28
输出: False
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
思路一:HashSet+DFS。如果HashSet中存在元素(k-val),那么就返回true,否则就把val放入HashSet中。
思路二:HashSet+BFS。同上,只不过遍历顺序不一样。
代码详解
思路一:HashSet+DFS
/**
* 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 {
HashSet<Integer> hs = new HashSet<>();
public boolean findTarget(TreeNode root, int k) {
if(root == null) {
return false;
}
if(hs.contains(k - root.val)) {
return true; // 存在就返回true
} else {
hs.add(root.val); // 不存在就add
}
return findTarget(root.left, k) || findTarget(root.right, k);
// ||符号可以在理想条件下不用执行后方的递归
}
}
思路二:HashSet+BFS(练手用)
class Solution {
HashSet<Integer> hs = new HashSet<>();
public boolean findTarget(TreeNode root, int k) {
// 经典BFS代码块
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()) {
TreeNode node = queue.poll();
if(hs.contains(k - node.val)) {
return true; // 存在就返回true,提前跳出循环
} else {
hs.add(node.val); // 不存在就add
}
// 非空孩子节点入队
if(node.left != null) {
queue.offer(node.left);
}
if(node.right != null) {
queue.offer(node.right);
}
}
return false;
}
}
注意点
- 本题没有利用BST的性质,通用性比较强。官方题解三是利用了BST性质,不过利用了性质也没降低题解的时间/空间复杂度。