题目
给了一个数组和一个目标值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;
}