给定一个整数数列,找出其中和为特定值的那两个数。
你可以假设每个输入都只会有一种答案,同样的元素不能被重用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
方法一
使用i,j遍历数组nums中的每一个变量,验证nums[i]+nums[j]是否等于target
代码如下:
public static int[] twoSum(int[] nums, int target) { int N = nums.length; for (int i=0; i<N-1; i++) for (int j=i+1; j<N; j++) if (nums[i] + nums[j] == target) return new int[] {i,j}; return null; }
此方法的复杂度为O(N^2)
方法二
使用HashMap存储nums的值及对应的下标
定义变量c=target-nums[i],在map中寻找是否存在键值为c
代码如下:
public static int[] twoSumMap(int[] nums, int target) { int N = nums.length; Map<Integer, Integer> map = new HashMap<>(); // nums的值为key,下标为value for (int i=0; i<N; i++) map.put(nums[i], i); for (int i=0; i<N-1; i++) { int complement = target - nums[i]; // 找到的键值不能为i if (map.containsKey(complement) && map.get(complement)!=i) return new int[] {i, map.get(complement)}; } return null; }
复杂度分析:
将nums存入map中,空间复杂度O(N)
for循环寻找两个下标,时间复杂度为O(N)
该方法中,键值为数组值,value为数组下标,故数组中数值相等的元素在map中只存储了一次。
如nums[] = {2, 5, 5, 6}, map中为2-0, 5-2, 6-3,但这并不影响该算法找到正确的答案
另外,若数组nums为有序的,则可以使用两个指针分别指向数组的首尾两个元素。
比较两个元素的和与target,根据比较结果移动指针,直到两指针相遇或找到结果
实现代码如下:
public static int[] twoSumOrder(int[] nums, int target) { int i = 0; int j = nums.length - 1; while (i != j) { if (nums[i] + nums[j] == target) return new int[] {i,j}; else if (nums[i] + nums[j] > target) j--; else i++; } return null; }
复杂度为O(N)