两数之和 三数之和【基础算法精讲 02】_哔哩哔哩_bilibili
两数之和 II - 输入有序数组
首先题目是一个已经排序好的数组,那么如图所示,可以发现如果选择3和8,加起来大于9,那么3和8中间的任何数与8加必然大于9,所以我们可以直接把8去掉,在剩下的数当中寻找答案。
因此在代码中,我们可以定义两个指针,初始位置分别在头尾,然后相加,如果和大于目标值,则尾指针向前移动一个,否则,头指针向后移动一个,直到找到答案。
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
left = 0
right = len(numbers)-1
while True:
s = numbers[left] + numbers[right]
if s == target:
break
if s > target:
right -= 1
else:
left += 1
return [left+1,right+1]
三数之和
与两数之和类似,我们可以先选取一个数字,然后在数组中进行匹配,尝试找到三数之和等于目标值的组合。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
res=[ ]
n=len(nums)
nums.sort()
if nums[0]>0 or n<3:
return []
for i in range(n):
if(nums[i]>0):
return res
if(i>0 and nums[i]==nums[i-1]):
continue
L=i+1
R=n-1
while(L<R):
if(nums[i]+nums[L]+nums[R]==0):
res.append([nums[i],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
L=L+1
R=R-1
elif(nums[i]+nums[L]+nums[R]>0):
R=R-1
else:
L=L+1
return res