您的位置:首页 > 其它

leetcode1---两数之和

2018-04-04 21:28 369 查看
给定一个整数数列,找出其中和为特定值的那两个数。

你可以假设每个输入都只会有一种答案,同样的元素不能被重用。

示例:

给定 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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: