题目:167.两数之和 II - 输入有序数组
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
- 返回的下标值(index1 和 index2)不是从零开始的。
- 你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
解题思路
- 用双指针:
1>.先定义2个指针,分别是左右指针L,R;
2>.令L=0,R=len(nums) -1,即分别指向数组的两端; - 判断 sum=nums[L]+nums[R] 的大小;
- sum==target, 符合要求,输出下标,之后,L+1,R-1;
- sum<target, L+1;
(因为数组是从小到大排序过的,当sum<0,两数之和整体来说是偏小的,自然要让两数中最小的数nums[L]变大,即指针L向右移动) - sum>target,R-1;
(同理,sum偏大,就让最大的值nums[R]变小)
代码(Python)
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
if(not numbers):
return []
L=0
R=len(numbers)-1
while(L<R):
sum=numbers[L]+numbers[R]
if(sum==target):
return [ L+1,R+1]
elif(sum>target):
R=R-1
else:
L= L+1
# return [-1,-1]
相信有的小伙伴已经发现了。当数组中有重复的元素时,其实是可以排除掉的。
更改代码如下:
代码(Python)
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
res=[]
if(not nums or len(nums)<3):
return []
nums.sort()
res=[]
for i in range(len(nums)-2):
#nums[i]都>0,三数都是正数,sum不可能=0
if(nums[i]>0):
return res
#排除重复的nums[i]
if(i>0 and nums[i]==nums[i-1]):
continue
L=i+1
R=len(nums)-1
while L<R:
sum = nums[i]+nums[L]+nums[R]
if sum ==0:
res.append([nums[i],nums[L],nums[R]])
L=L+1
R=R-1
elif sum < 0:
L=L+1
else:
R=R-1
#排除重复的nums[L],nums[R]
while(L<R and nums[L]==nums[L+1]):
L=L+1
while(L<R and nums[R]==nums[R-1]):
R=R-1
return res
结果
总结
好了,这是一道双指针的简单问题,其实,还有很多其他解法,比如哈希表,二分查找!有兴趣的同学可以去找找。完结撒花~
后续
同时,贴上咱的个人博客,欢迎客官大老爷们来访~
青枫阁