题目:
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出和为目标值 target 的那两个整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
解题思路:
一开始惯性思维,想法就是二重循环。设计出来的算法简单易实现,但是效率、内存消耗等都很不如意,时间效率是O(n^2),其中n是数组元素数量,最坏情况下数组中任意两个数都要被匹配一次。代码如下
class Solution {
public int[] twoSum(int[] nums, int target) {
int length=nums.length,i,j;
for (i=0;i<length-1;i++){
for (j=i+1;j<length;j++){
if (nums[i]+nums[j]==target){
return new int[]{i,j};
}
}
}
return null;
}
}
经过思考,发现有另一种更优的算法。注意到上一个方法的时间复杂度较高的原因是寻找 target - nums[i] 的时间复杂度过高。因此,我们需要一种更优秀的方法,能够快速寻找数组中是否存在目标元素。如果存在,我们需要找出它的索引。考虑使用哈希表,可以将寻找 target - nums[i] 的时间复杂度降低到O(1)。这样我们创建一个哈希表,对于每一个 nums[i],我们首先查询哈希表中是否存在 target - nums[i],然后将 nums[i] 插入到哈希表中,即可保证不会让 x 和自己匹配。代码实现如下
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];
}