你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
提示:给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
分析:
数组中同一个元素不能使用两遍,那么暴力解题法(双for循环)就不适用了,而且这种方法时间复杂度也比较高,这里考虑使用HashMap来解题。
思路
在for循环里面,求target和每次循环的数组值之差,在HashMap里get这个差,如果不能拿到,就把当前数组值和数组下标分别作为Key和Value存放到HashMap中;如果拿到了,可直接返回当前差在Map中对应的值和当前数组下标即可。
根据HashMap的特性,根据key寻找value的时间复杂度为O(1),因此效率很高。
与暴力解题法执行时间对比
第二行是用暴力解题法,可以看到第一种大大节省了运行时间。
示例代码
public int[] twoSum(int[] nums, int target) {
Map m = new HashMap<Integer, Integer>();
for(int i=0; i<nums.length; i++) {
Integer a = target - nums[i];
if(m.get(a) == null) {
m.put(nums[i],i);
}else {
int[] aa = {(int)m.get(a), i};
return aa;
}
}
return null;
// 暴力解题法 不推荐
// for(int i=0; i<nums.length - 1; i++) {
// for(int j=i+1; nums.length>j ; j++) {
// int sum = nums[i] + nums[j];
// if(sum == target) {
// int[] a = {i,j};
// return a;
//
// }
// }
// }
// return null;
}