浅谈Hash Map中解题技巧之先查再添加元素的经典应用

在使用HashMap解题时经常需要判断数组序列中是否存在两个值之和等于某一个目标值的题目, 在解答这类题目的一个典型暴力做法：

双重for循环进行查找：

for (int i = 0; i < n; i++) {
   for (int j= i + 1; i < n; j++) {
       // nums[i] + nums[j] 是否等于目标值target
   }
}

这样时间复杂度太高了，需要进行优化，我们先假设数组nums中存在两个数的下标 i 与 j，对应的数据nums[i] + nums[j] == target, 此时我们定义一个空的HashMap对象 map 依次将 nums 数组中元素存入map当存入第 i 个数据时，此时 target - nums[i] 肯定不在 map 中，将nums[i] 正常存入map,当存入第 j 个数据 nums[j] 时， 此时再判断 target - nums[j] 也就是之前存入的nums[i] 肯定在map中存在，即可判断存在两个不同的元素相加等于目标值；

注意，在这个过程中遍历数组每个元素时，先判断map中是否存在 target - nums[curIndex] 如果存在直接返回True, 反之将当前遍历值存入map, 这样也可以避免同一个元素被重复计算。 

样题1：

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

提示：

• 2 <= nums.length <= 104
• -109 <= nums[i] <= 109
• -109 <= target <= 109
• 只会存在一个有效答案

进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？

import java.util.HashMap;
import java.util.LinkedHashMap;
class Solution {
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer, Integer> keyMap = new LinkedHashMap<>();
        for (int i = 0; i < nums.length; i++) {
            if (keyMap.containsKey(target - nums[i])) {
                return new int[]{keyMap.get(target - nums[i]), i};
            }
            keyMap.put(nums[i], i);
        }
        return new int[0];
    }
}

样题2：

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

示例 1：

输入: root = [5,3,6,2,4,null,7], k = 9
输出: true

示例 2：

输入: root = [5,3,6,2,4,null,7], k = 28
输出: false

提示:

• 二叉树的节点个数的范围是  [1, 104].
• -104 <= Node.val <= 104
• 题目数据保证，输入的 root 是一棵 有效 的二叉搜索树
• -105 <= k <= 105

/**
 * 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> set = new HashSet<>();
    public boolean findTarget(TreeNode root, int k) {
        if (root == null) {
            return false;
        }
        if (set.contains(k - root.val)) {
            return true;
        }
        set.add(root.val);
        return findTarget(root.left, k) || findTarget(root.right, k);
    }
}