題目描述:
給定一個二叉搜索樹和一個目標結果,如果 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);
}