题目描述
给定一个整数数列,找出其中和为特定值的那两个数。
你可以假设每个输入都只会有一种答案,同样的元素不能被重用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解题思路和方法
方法一:蛮力解决
遍历每个元素x,并确定是否有另外一个元素y=target-x。
时间复杂度分析
时间复杂度为N*N,因为有内外双重循环;
空间复杂度为O(1);
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");
}
方法二:两次HashMap
第一次遍历数组每个元素x,将<x,x的索引>存入HashMap;
第二次遍历数组每个元素x,在HashMap中查找是否存在target-x,如果存在并且不是x本身,就返回结果;
时间复杂度分析
时间复杂度为O(n),两次遍历数组,并且HashMap的查找时间开销为O(1);
空间复杂度为O(n),这取决于HashMap的空间开销;
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");
}
方法三:一次HashMap
就进行一次遍历数组每个元素x,在把<x,x的索引>存入HashMap之前,检查HashMap是否存在target-x;
为什么要在存入<x,x的索引>之前检查HashMap是否存在target-x?这是为了避免target-x == x的情况出现;
时间复杂度分析
时间复杂度为O(n),一次数组遍历;空间复杂度为O(n),这取决于HashMap的空间开销;
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i]; //
if (map.containsKey(complement)) { //
return new int[] { map.get(complement), i };
}
map.put(nums[i], i); //
}
throw new IllegalArgumentException("No two sum solution");
}