题目:输入一个递增排序的数组和一个数字s,在数组中查找两个数,使得它们的和正好是s。如果有多对数字的和等于s,则输出任意一对即可。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[2,7] 或者 [7,2]
示例 2:
输入:nums = [10,26,30,31,47,60], target = 40
输出:[10,30] 或者 [30,10]
限制:
1 <= nums.length <= 10^5
1 <= nums[i] <= 10^6
方法一:Map
Map中key:target - nums[i] value:i
通过containsKey方法可以知道当前元素是否为之前已经存在的元素对应的差值,如果是则直接返回结果,否则继续找。
代码:
class Solution {
public int[] twoSum(int[] nums, int target) {
// map k:target - nums[i] v:i
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(nums[i])) return new int[]{nums[map.get(nums[i])], nums[i]};
map.put(target - nums[i], i);
}
return null;
}
}
时间复杂度:O(n) 遍历了数组
空间复杂度:O(n) 会将找到结果目标元素之前的所有元素的对应差存起来
方法二:双指针
双指针i = 0, j = nums.length - 1
由于数组是递增排序的,那么,可以计算nums[i] + nums[j] 和target的关系,如果大了,那就j–,小了i++,相等则直接返回i,j
代码:
class Solution {
public int[] twoSum(int[] nums, int target) {
// 递增 可以用双指针i j i = 0, j = length - 1
int len = nums.length;
int i = 0;
int j = len - 1;
int sum = 0;
while (i < j) {
sum = nums[i] + nums[j];
if (sum == target) return new int[]{nums[i], nums[j]};
else if (sum > target) j--;
else i++;
}
return null;
}
}
时间复杂度:O(n)
空间复杂度:O(1)