题目信息
-
时间: 2019-07-05
-
题目链接:Leetcode
-
tag: 双指针 哈希表
-
难易程度:简单
-
题目描述:
输入一个递增排序的数组和一个数字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. 1 <= nums.length <= 10^5
2. 1 <= nums[i] <= 10^6
解题思路
本题难点
查找和为s的两个数字,性能最优。
具体思路
本题的 nums 是 排序数组 ,因此可使用 双指针法 将空间复杂度降低至 O(1)。
代码
class Solution {
public int[] twoSum(int[] nums, int target) {
//双指针 i , j 分别指向数组 nums 的左右两端
int i = 0;
int j = nums.length - 1;
while(i < j){
//计算和 s=nums[i]+nums[j]
int sum = nums[i] + nums[j];
//若 s>target ,则指针 j 向左移动,即执行 j=j−1
if(sum > target){
j--;
}else if(sum < target){//若 s<target ,则指针 i 向右移动,即执行 i=i+1
i++;
}else{//若 s=target ,立即返回数组 [nums[i],nums[j]]
return new int[]{nums[i],nums[j]};
}
}
return new int[0];
}
}
复杂度分析:
- 时间复杂度 O(N) : N为数组 nums 的长度;双指针共同线性遍历整个数组。
- 空间复杂度 O(1) : 变量 i, j 使用常数大小的额外空间。
其他优秀解答
解题思路
哈希表查找,利用 HashMap 可以通过遍历数组找到数字组合,时间和空间复杂度均为 O(N);
代码
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer,Integer> map=new HashMap();
for(int i=0;i<nums.length;i++){
map.put(target-nums[i],nums[i]);
}
for(int i=0;i<nums.length;i++){
//注意排除当前元素
if(map.containsKey(nums[i]) && map.get(nums[i])!=i){
return new int[]{nums[i],map.get(nums[i])};
}
}
return new int[0];
}
}