给定一个整数数列,找出其中和为特定值的那两个数。
你可以假设每个输入都只会有一种答案,同样的元素不能被重用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路:
在一个数组中查找两数之和等于一个值,首先想到的应该是选择数组其中一个数,然后查找另外一个数,思路有排序后二分查找,使用hash查找。
但是这题的不同点在于要求查找的是下标,所以无论使用哪种方法去查找,要求要保存现场,常用思路是将数值与下标进行绑定然后进行查找,用二分查找比较容易实现。
我这里使用的是hash查找,hash查找的话将数值与下标进行绑定,数值为key,下标为value,由于数组存在重复值,所以需要对数据进行过滤。两值之和等于指定值,在hashmap中插入数据时,先判断值是否存在,如果不存在,直接插入;如果存在,判断数值*2是否等于target,相等,则输出两个数下标,否则继续查找。重复操作将数据存入hashmap。
然后进行遍历查找,需要注意的是,存在value<<1=target的情况,因为查找之前过滤过数据,所以此处需要特判一下,防止出现只有一个value,但value<<1=target从而输出两个相等下标的情况。
代码如下:
public int[] twoSum(int[] nums, int target) {
int[] rs = new int[2];
HashMap<Integer, Integer> numMap = new HashMap<>(1024);
for (int i=0; i<nums.length; i++){
if (numMap.containsKey(nums[i])){
if (nums[i]<<1 == target){
rs[0] = numMap.get(nums[i]);
rs[1] = i;
return rs;
}
continue;
}
numMap.put(nums[i], i);
}
for (int i=0; i<nums.length; i++){
if (numMap.containsKey(target-nums[i]) && nums[i]<<1 != target){
rs[0] = i;
rs[1] = numMap.get(target-nums[i]);
break;
}
}
return rs;
}