来看一下题目,给定一个整数数组,另外给定一个数,如果在数组中两个数相加等于这个数,获得这两个数的索引,两个数不能重复,而且假设每个数最多只有一组数存在于数组中。
Example:
Given nums = [2, 7, 11, 15], target = 9, Because nums[0] + nums[1] = 2 + 7 = 9, return [0, 1].
假设已经给出 nums 和target, 写出其对应的算法。
第一种方法,暴力破解
public int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[j] == target - nums[i]) {
return new int[] { i, j };
}
}
}
throw new IllegalArgumentException("No two sum solution");
}
这个算法的过程,遍历整个数组,内循环为当前值向后遍历,故其时间复杂度为o(n^2), 效率比较低。
第二种方法,利用哈希表,key-value对 ,这里需要注意的是,nums[i]作为key,i 作为value,因为我们需要得到索引,但map 中没有根据value查询对应key的方法。
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
map.put(nums[i], i);
}
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i];
if (map.containsKey(complement) && map.get(complement) != i) {
return new int[] { i, map.get(complement) };
}
}
throw new IllegalArgumentException("No two sum solution");
}
第一个for,初始化map, 第二个for,遍历nums,相减,从map需要找的另一半,如果有,并且不是它自身,那么说明找到了。这个算法的时间复杂度为o(n),相较于上一个算法效率提升不少。但是还是美中不足,还能改进。
第三种方法, 第二种方法的基础上进改进。
public static int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for(int i=0; i<nums.length; i++){
int temp = target - nums[i];
if(map.containsKey(temp)){
return new int[]{map.get(temp), i};
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
在把nums[i]放到map之前,判断map是否包含它的另一半, 如果包含就找到了,如果不包含,就继续添加,添加继续判断,直到把所有的nums添加到map中,这个虽然时间复杂度也是o(n),但是运行时间却比方法二少了很多。