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

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

Example 1:

Input: root = [5,3,6,2,4,null,7], k = 9
Output: true

Example 2:

Input: root = [5,3,6,2,4,null,7], k = 28
Output: false

Example 3:

Input: root = [2,1,3], k = 4
Output: true

Example 4:

Input: root = [2,1,3], k = 1
Output: false

Example 5:

Input: root = [2,1,3], k = 3
Output: true

Constraints:

• The number of nodes in the tree is in the range [1, 104].
• -104 <= Node.val <= 104
• root is guaranteed to be a valid binary search tree.
• -105 <= k <= 105

两数之和 IV - 输入 BST。

给定一个二叉搜索树 root 和一个目标结果 k，如果 BST 中存在两个元素且它们的和等于给定的目标结果，则返回 true。

这道题不难，这里我提供 BFS 和 DFS 两种做法，两种做法时间空间复杂度都一样。这道题其实还是找两数之和，无非是在一棵树里找。因为遍历的是树所以我们不能排序，但是我们还是可以利用 hashset 来提速。对于当前遇到的节点值 root.val，如果 hashset 里存在另一个值 K - root.val 则返回 true。遍历完整棵树还是找不到的话，则返回 false。

时间O(n)

空间O(n)

BFS实现

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

 

DFS实现

/**
 * 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) {
        HashSet<Integer> set = new HashSet<>();
        return dfs(root, set, k);
    }

    private boolean dfs(TreeNode root, HashSet<Integer> set, int k) {
        if (root == null) return false;
        if (set.contains(k - root.val)) {
            return true;
        }
        set.add(root.val);
        return dfs(root.left, set, k) || dfs(root.right, set, k);
    }
}

 

LeetCode 题目总结