LeetCode-653. Two Sum IV - Input is a BST

653. Two Sum IV - Input is a BST

Given a Binary Search Tree and a target number, return true if there exist two elements in the BST such that their sum is equal to the given target.

Example 1:

Input: 
    5
   / \
  3   6
 / \   \
2   4   7

Target = 9

Output: True

Example 2:

Input: 
    5
   / \
  3   6
 / \   \
2   4   7

Target = 28

Output: False

思路：

下面有三种解法：

其中第一种和第二种都是在遍历的过程中来将二叉树的节点存入 set 中，一边遍历一边来寻找符合条件的节点（k-root.val 和 root.val）。使用 set 类可以直接使用contains方法。

• 其中第一种是BFS和非递归方法实现的
• 第二种是直接遍历和递归方法实现的

第三种方法：先使用中序遍历，得到一个升序排列的list，再while寻找满足条件的节点。

不过没想到第一种是最慢的。。。

第一种方法：

/**
 * 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){
        Set<Integer> set =new HashSet();
        Queue<TreeNode> queue = new LinkedList();
        queue.add(root);
        while(!queue.isEmpty()){
            if(queue.peek()!=null){
                TreeNode node = queue.remove();
                if(set.contains(k-node.val)) return true;
                set.add(node.val);
                queue.add(node.right);
                queue.add(node.left);
            }else queue.remove();
            
        }
        return false;
    }
}

Runtime: 34 ms

第二种方法：

class Solution {
    public boolean findTarget(TreeNode root,int k){
        Set<Integer> set =new HashSet();
        return find(root,k,set);
    }
    
    public boolean find(TreeNode root,int k,Set<Integer> set){
        if(root == null) return false;
        if(set.contains(k-root.val)) return true;
        set.add(root.val);
        return find(root.left,k,set)||find(root.right,k,set);
    }
}

Runtime: 29 ms

第三种方法：

class Solution {
    public boolean findTarget(TreeNode root,int k){
        List<Integer> list = new ArrayList();
        inorder(root,list);
        int l =0,r = list.size()-1;
        while(l<r){
            int sum = list.get(l) + list.get(r);
            if(sum == k) return true;
            if(sum <k) l++;
            else r--;
        }
        return false;
        
    }
    
    public void inorder(TreeNode root,List<Integer> list){
        if(root == null) return ;
        inorder(root.left,list);
        list.add(root.val);
        inorder(root.right,list);
    }
}

Runtime: 29 ms