229、兩數之和IV

題目描述:
給定一個二叉搜索樹和一個目標結果,如果 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);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章