剑指 Offer 57. 和为s的两个数字
链接:https://leetcode-cn.com/problems/he-wei-sde-liang-ge-shu-zi-lcof/
来源:力扣(LeetCode)
一、题干
输入一个递增排序的数组和一个数字s,在数组中查找两个数,使得它们的和正好是s。如果有多对数字的和等于s,则输出任意一对即可。
二、示例
输入:nums = [2,7,11,15], target = 9
输出:[2,7] 或者 [7,2]
输入:nums = [10,26,30,31,47,60], target = 40
输出:[10,30] 或者 [30,10]
三、提示
1 <= nums.length <= 10^5s
1 <= nums[i] <= 10^6
四、解答
解答一 思路:HashMap
class Solution {
public int[] twoSum(int[] nums, int target) {
/* hashmap解法,力扣上时间13%,空间5%,更适用于未排序数组 */
Map<Integer, Integer> map = new HashMap<>();
for(int num : nums){
if(map.containsKey(num)) return new int[]{num, map.get(num)};
else map.put(target - num, num);
}
return null;
}
}
对于每一个数字num,先看一看map的Key里有没有,如果有的话就是答案,如果没有的话就令target-num为key,num为value放进map中。时间复杂度:O(n),空间复杂度:O(n)。
解答二 思路:双指针
class Solution {
public int[] twoSum(int[] nums, int target) {
/* 双指针解法,力扣上时间97%,空间62%,适用于排序数组 */
int i = 0, j = nums.length - 1;
while(i < j){
if(nums[i] + nums[j] < target) i++;
else if(nums[i] + nums[j] > target) j--;
else return new int[]{nums[i], nums[j]};
}
return null;
}
}
前后两个指针,遍历时,当 nums[i] + nums[j] < target 时,说明两个数的和小了,i向前,当 nums[i] + nums[j] > target 时,说明两个数的和大了,j向后。由于是排序数组,理论上只要存在答案就一定可以找到。时间复杂度:O(n),空间复杂度:O(1)。
五、后记
建议看这位大佬作者:jyd的题解,更清晰明了。
链接:https://leetcode-cn.com/problems/he-wei-sde-liang-ge-shu-zi-lcof/solution/mian-shi-ti-57-he-wei-s-de-liang-ge-shu-zi-shuang-/
来源:力扣(LeetCode)