LeetCode - Two Sum

题目

　　给了一个数组和一个目标值target，求数组中哪两个数的和等于target，返回坐标。

Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

思路

　　粗暴的解决办法就是遍历，时间复杂度为O(n!)。这样的思路是，先找到一个数，再去遍历查找满足和为target的数，既然这样，我们可以将数组的信息保存下来，将值作为key，坐标作为value保存到HashMap中，这样可以用O(1)的复杂度找到要找的数。
　　碰到两个相同的数，比如输入[2,3,2,4]和4，在查找的时候，第二次碰到2的时候，从map中取出2对应的value，加入结果集中返回即可。
　　如果输入的是[2,3,2,4]和6，当遍历到第二个2时，因为map中已经存在，所以就不需要更新了。

代码

public int[] twoSum(int[] nums, int target) {
    int[] res = new int[2];
    if (nums == null || nums.length < 2) {
        return res;
    }
    HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
    for (int i = 0 ; i < nums.length ; i++) {
        // 在map中找有没有和当前值加起来等于target的值，有的话就返回
        if (map.get(target - nums[i]) != null) {
            res[0] = map.get(target - nums[i]);
            res[1] = i;
            return res;
        }
        if (map.get(nums[i]) == null) {
            // 将值作为key，坐标作为value
            map.put(nums[i], i);
        }
    }
    return res;
}