leetcode 1. 两数之和（Java版）

题目描述：（题目难度，简单）

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

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

示例代码

下面给出时间复杂度为，空间复杂度为的Java代码（可以直接在 leetcode 上提交）：

class Solution {
    public int[] twoSum(int[] nums, int target) {//前两行是 leetcode 提供的模板
        int[] res = {0,0};
        HashMap hashMap = new HashMap<>();
        for(int i = 0; i < nums.length; ++i){
            if(hashMap.size() > 0){
                if(hashMap.containsKey(target - nums[i])){
                    res[0] = hashMap.get(target - nums[i]);
                    res[1] = i;
                    return res;
                }
            }
            hashMap.put(nums[i], i);
        }
        return res;
    }
}

思路解析

在遍历数组的过程中，将遍历过的元素都扔到 HashMap 当中。当遍历到当前元素时，查找目标值与当前元素的差值是否在 HashMap 中，如果存在，则算法结束，如果不存在，则将当前元素加入到 HashMap 中，然后遍历下一元素。
为什么能这么做？因为如果输入有解的话，那么要找的那两个元素必定一前一后，当我们遍历到后一元素时，前一元素必定在 HashMap 中，即查找一下 HashMap 便可以得到答案。