- 两数之和 IV - 输入 BST
要求:给定一个二叉搜索树 root 和一个目标结果 k,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
代码:
/**
* 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 {
Map<Integer, Integer> map = new HashMap<>();
public boolean findTarget(TreeNode root, int k) {
if(root == null){
return false;
}
if(map.containsKey(k - root.val)){
return true;
}
map.put(root.val, 1);
return findTarget(root.left, k) || findTarget(root.right, k);
}
}