题目描述:
给定一个二叉搜索树和一个目标结果,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
案例 1:
输入:
5
/
3 6
/ \
2 4 7
Target = 9
输出: True
案例 2:
输入:
5
/
3 6
/ \
2 4 7
Target = 28
输出: False
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean findTarget(TreeNode root, int k) {
// 第一种思路就是转成list放入然后判断是否有二者之和
List<Integer> list = new ArrayList<>();
get(list,root);
for (Integer integer : list) {
if(k - integer != integer){
if(list.contains(new Integer(k - integer))){
return true;
}
}
}
return false;
}
public void get(List<Integer> list,TreeNode node){
if(node == null){
return ;
}else {
list.add(node.val);
get(list,node.left);
get(list,node.right);
}
}
}
当然这种方法比较笨拙,但是能够完善
他这里使用的是hashset来完善,觉得这个比我的list效果要好一些。
看看别人完善的思路:
public boolean findTarget(TreeNode root, int k) {
HashSet<Integer> hashset = new HashSet<Integer>();
return preOrder(root,hashset,k);
}
public boolean preOrder(TreeNode root,HashSet<Integer> hashset,int k){
if(root == null)
return false;
if(hashset.contains(k - root.val)){
return true;
}
hashset.add(root.val);
return preOrder(root.left,hashset,k) || preOrder(root.right,hashset,k);
}