【LeetCode】第653题——两数之和 IV - 输入 BST（难度：简单）

题目描述

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

1. 示例 1:
   输入:

        5
       / \
      3   6
     / \   \
    2   4   7
   

   Target = 9
   输出: True

2. 示例 2:
   输入:

        5
       / \
      3   6
     / \   \
    2   4   7
   

   Target = 28
   输出: False

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

思路一：HashSet+DFS。如果HashSet中存在元素(k-val)，那么就返回true，否则就把val放入HashSet中。

思路二：HashSet+BFS。同上，只不过遍历顺序不一样。

代码详解

思路一：HashSet+DFS

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {

    HashSet<Integer> hs = new HashSet<>();

    public boolean findTarget(TreeNode root, int k) {
        if(root == null) {
            return false;
        }
        if(hs.contains(k - root.val)) {
            return true;		// 存在就返回true
        } else {
            hs.add(root.val);	// 不存在就add
        }
        return findTarget(root.left, k) || findTarget(root.right, k);
        // ||符号可以在理想条件下不用执行后方的递归
    }
}

思路二：HashSet+BFS（练手用）

class Solution {

    HashSet<Integer> hs = new HashSet<>();

    public boolean findTarget(TreeNode root, int k) {
    	// 经典BFS代码块
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()) {
            TreeNode node = queue.poll();
            if(hs.contains(k - node.val)) {
                return true;		// 存在就返回true，提前跳出循环
            } else {
                hs.add(node.val);	// 不存在就add
            }
            // 非空孩子节点入队
            if(node.left != null) {
                queue.offer(node.left);
            }
            if(node.right != null) {
                queue.offer(node.right);
            }
        }
        return false;
    }
}

注意点

• 本题没有利用BST的性质，通用性比较强。官方题解三是利用了BST性质，不过利用了性质也没降低题解的时间/空间复杂度。