LeetCode 001 Two Sum

Two Sum

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

题目大意及分析

这是leetcode的第一个题，是简单级别的。

大意就是给出一个数组，给出一个目标数字，在数组中选两个不同的数字相加得目标数字则完成。返回两个数字的下标。且保证有且只有一个答案。

一开始我的想法就是很简单的两次循环，这个肯定大多数人都能想到，就不想分析了。然后说一下我看到的一个 solution 吧，使用的 map，value值 是数值大小，key值 是数组下标。一次循环，每次用 ( target - value) 值在 map 中寻找，找得到并且key值和当前值不同则输出，找不到进入下一次循环。

Map还是很好用的啊，原来用的很少，这次学到了。

代码

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        //输入map
    for (int i = 0; i < nums.length; i++) {
        map.put(nums[i], i);
    }
    for (int i = 0; i < nums.length; i++) {
        int complement = target - nums[i];//按照需要的value值去寻找
        if (map.containsKey(complement) && map.get(complement) != i) {
            return new int[] { i, map.get(complement) };
        }
    }
    throw new IllegalArgumentException("No two sum solution");
    }
}