给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例3:
输入:nums = [3,3], target = 6
输出:[0,1]
刚读题说真的还是很懵逼,于是我点开了三个英语的提示看了一看。概括每个点如下所示:
1)暴力解法,最傻瓜的想法
2)我们选定一个元素x,那么我们就需要在数组中寻找到另外一个元素y,y=target-x。我们能不能改变这个数组,让这种x,y的对应关系的搜索变得更加快呢?
3)另外一个思路是,我们不去改变数组,而尝试“增加空间”,我们可以使用hashmap的结构实现。
其实想想也是,题干中已经提示了只会存在一个有效答案这样的提示,很明显是有意向map结构引导。
上面的提示总的概括一下,那就是用一个hashmap存储符合下述映射的键值对:数组元素值与数组坐标。然后我们再一次遍历数组,找到满足下述关系的当前遍历数组元素(我们叫它为A)+键值对组合:A+键值对的key=target。然后我们获取键值对的value以及A的下标(也就是当前循环变量值),这就是我们要的结果。
我的解答如下:
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<>();
for(int i = 0;i<nums.length;i++){
map.put(target-nums[i],i);
}
int index1 = 0;
int index2 = 0;
for (int i = 0;i<nums.length;i++){
if(map.containsKey(nums[i])){
index1 = i;
index2 = map.get(nums[i]);
if(index1==index2)
continue;//这是避免数组中出现值为target/2的元素的情形
break;
}
}
return new int[]{index1,index2};
}
}
示例2和示例3说明了当数组中有一个元素,其值=target/2时,会出现index1=index2的情况。因此我们还要加一个条件判断。
不过示例3的情况会导致map只能最终存储一个键值对,虽然在测试时还是可以输出正确结果…#
官方题解
暴力解法的嵌套循环导致时间复杂度为
O
(
N
)
O(N)
O(N)。算法的时间复杂度较高的原因是寻找 target - x 的时间复杂度过高。因此,我们需要一种更优秀的方法,能够快速寻找数组中是否存在目标元素。如果存在,我们需要找出它的索引。
使用哈希表,可以将寻找 target - x 的时间复杂度降低到从 O(N)(需要遍历全表查找)降低到 O(1)。正如前面所做的题所言,哈希表的查询是非常快的。如果需要提升查找元素的速度的话,首先就要想到使用HashMap。
这样我们创建一个哈希表,对于每一个 x,我们首先查询哈希表中是否存在 target - x,然后将 x 插入到哈希表中,即可保证不会让 x 和自己匹配。
以下是代码:
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> hashtable = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; ++i) {
if (hashtable.containsKey(target - nums[i])) {
return new int[]{hashtable.get(target - nums[i]), i};
}
hashtable.put(nums[i], i);
}
return new int[0];//返回空数组
}
}
我的解答与官方解答的差异:
- 方法中的for循环可以直接使用return代替break。我没有使用break。
- 完全可以一边向哈希表中存数据,一边查询哈希表中是否存在符合target-nums[i]的键值对。因为我们只需要找到一个两数之和的组合,只要找到一个就可以立刻停止查找。(这一点我也想到了,但是没有深究。应该细想的)。而且这种做法还完全规避了示例3中需要存入两个key值相等的键值对的情况,是更好的。
收获
我认为这题如果在之后接触了大量使用哈希表处理数据的情形下可以很快做出来。想到用hashmap实际上不需要太多基础,但是怎么用hashmap才是体现功力深厚的地方