给定一个二叉搜索树 root 和一个目标结果 k,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
示例 1:
输入: root = [5,3,6,2,4,null,7], k = 9
输出: true
提示:
- 二叉树的节点个数的范围是 [1, 104].
- -104 <= Node.val <= 104
- root 为二叉搜索树
- -105 <= k <= 105
解法
class Solution {
public boolean findTarget(TreeNode root, int k) {
return doFindTarget(root, root, k);
}
private boolean doFindTarget(TreeNode root, TreeNode tree, int k) {
if (tree == null) {
return false;
}
if (search(root, tree, k - tree.val)) {
return true;
}
if (tree.left != null && doFindTarget(root, tree.left, k)) {
return true;
}
if (tree.right != null && doFindTarget(root, tree.right, k)) {
return true;
}
return false;
}
private boolean search(TreeNode root, TreeNode use, int k) {
if (root == null) {
return false;
}
if (root.val == k && root != use) {
return true;
}
if (root.val >= k) {
return search(root.left, use, k);
} else {
return search(root.right, use, k);
}
}
}