前言
跟着灵茶山艾府大佬学算法!
两数之和 三数之和【基础算法精讲 01】_哔哩哔哩_bilibili
使用相向双指针的前提:
数组已经排序完毕
提示:以下是本篇文章正文内容,下面案例可供参考
一、Leetcode167. 两数之和 II - 输入有序数组
给你一个下标从 1 开始的整数数组 numbers
,该数组已按 非递减顺序排列 ,请你从数组中找出满足相加之和等于目标数 target
的两个数。如果设这两个数分别是 numbers[index1]
和 numbers[index2]
,则 1 <= index1 < index2 <= numbers.length
。
以长度为 2 的整数数组 [index1, index2]
的形式返回这两个整数的下标 index1
和 index2
。
你可以假设每个输入 只对应唯一的答案 ,而且你 不可以 重复使用相同的元素。
你所设计的解决方案必须只使用常量级的额外空间。
示例 1:
输入:numbers = [2,7,11,15], target = 9 输出:[1,2] 解释:2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。
示例 2:
输入:numbers = [2,3,4], target = 6 输出:[1,3] 解释:2 与 4 之和等于目标数 6 。因此 index1 = 1, index2 = 3 。返回 [1, 3] 。
示例 3:
输入:numbers = [-1,0], target = -1 输出:[1,2] 解释:-1 与 0 之和等于目标数 -1 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。
解题思路 使用相向双指针
left=0,right=len(numbers),s = numbers[left] + numbers[right]
因为数组本身已经按照从小到大排列好顺序了,因此如果s>target,则可以通过right = right - 1调整使s变小,如果s < target,则通过left = left + 1,使s变大。 直到找到s == taget。
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
left, right = 0, len(numbers)-1
while left < right:
s = numbers[left] + numbers[right]
if s == target:
return left+1, right+1
elif s > target:
right = right - 1
else:
left = left + 1
二、Leetcode15. 三数之和
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0
且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]] 解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。 nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。 nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。 不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1] 输出:[] 解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0] 输出:[[0,0,0]] 解释:唯一可能的三元组和为 0 。
解题思路 方法仍然是使用相向双指针
要求:现在是三数之和,要求是三个数相加的和等于0,并且这三个数的下标不能相同且不能重复出现相同的三元组。
按住一个数不动,然后去枚举第二个数和最后一个数进行相向双指针。 这样做就可以将包含第一个数的满足要求的三元组全部求出。
然后依次按住第二个数、第三个数....一直到倒数第三个数。因为三元组要求包含三个数,所以取到倒数第三个数就可以停下来了。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
nums.sort()
i = 0
n = len(nums)
ans = []
while i < n - 2:
# 为了防止出现重复的三元组
if i > 0 and nums[i] == nums[i-1]:
i = i + 1
continue
# 优化1: 三个最小的数相加都>0,没必要继续循环了,后续的数怎么加都>0。
if nums[i] + nums[i+1] + nums[i+2] > 0:
break
# 优化2:如果当前的数加上最后两个数,仍然<0,那么在[i,n-1]这个范围随便加都<0,故可以跳出。
if nums[i] + nums[n-1] + nums[n-2] < 0:
i = i + 1
continue
j = i + 1
k = n - 1
while j<k:
s = nums[i] + nums[j] + nums[k]
if s > 0:
k = k - 1
elif s < 0:
j = j + 1
else:
ans.append([nums[i], nums[j], nums[k]])
j = j + 1
k = k - 1
# 去重
while j<k and nums[j] == nums[j-1]:
j = j + 1
# 去重
while k>j and nums[k] == nums[k+1]:
k = k - 1
i = i + 1
return ans
三、Leetcode16. 最接近的三数之和
给你一个长度为 n
的整数数组 nums
和 一个目标值 target
。请你从 nums
中选出三个整数,使它们的和与 target
最接近。
返回这三个数的和。
假定每组输入只存在恰好一个解。
示例 1:
输入:nums = [-1,2,1,-4], target = 1 输出:2 解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
示例 2:
输入:nums = [0,0,0], target = 1 输出:0
解题思路 这题与上一题基本是一样 都是要按住一个数,然后枚举后两个数!
只是这题要求返回的是和,所以不需要去重,因为返回的是一个数!
最接近 可以用距离来描述 abs(s-target) 这个数越小,说明越接近。
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
nums.sort()
n = len(nums)
distance = float('inf')
ans = 0
for i in range(n):
j = i + 1
k = n - 1
while j < k:
s = nums[i] + nums[j] + nums[k]
if abs(s - target) < distance:
ans = s
distance = abs(s - target)
if s > target:
k = k - 1
elif s < target:
j = j + 1
else:
return s
return ans
四、Leetcode18. 四数之和
给你一个由 n
个整数组成的数组 nums
,和一个目标值 target
。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]]
(若两个四元组元素一一对应,则认为两个四元组重复):
0 <= a, b, c, d < n
a
、b
、c
和d
互不相同nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。
示例 1:
输入:nums = [1,0,-1,0,-2,2], target = 0 输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2:
输入:nums = [2,2,2,2,2], target = 8 输出:[[2,2,2,2]]
解题思路 其实和三数之和的做法基本是一样的,只是这次要按住的是两个数。 然后对后两个数使用相向双指针。
有一些小细节需要注意
- 按住 i 的时候,j的起始位置应该是i+1。
- 判断j是否重复的时候,判断条件应该是 j > i+1,当
j
刚开始等于i+1
的时候,如果直接执行if nums[j] == nums[j-1]:
,会导致在j
第一次移动时,就检查与前一个元素是否相等,从而可能跳过一些潜在的解。通过添加j > i+1
的条件,确保了在j
移动一次之后才会开始检查是否相等,避免了这个问题。
class Solution:
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
nums.sort()
n = len(nums)
ans = []
for i in range(n-3):
if i > 0 and nums[i] == nums[i-1]:
continue
for j in range(i+1, n-2):
if j > i+1 and nums[j] == nums[j-1]:
continue
k = j + 1
v = n - 1
while k < v:
s = nums[i] + nums[j] + nums[k] + nums[v]
if s > target:
v = v - 1
elif s < target:
k = k + 1
else:
ans.append([nums[i], nums[j], nums[k], nums[v]])
k = k + 1
v = v - 1
while k < v and nums[k] == nums[k-1]:
k = k + 1
while v > k and nums[v] == nums[v+1]:
v = v - 1
return ans
五、Leetcode2824. 统计和小于目标的下标对数目
给你一个下标从 0 开始长度为 n
的整数数组 nums
和一个整数 target
,请你返回满足 0 <= i < j < n
且 nums[i] + nums[j] < target
的下标对 (i, j)
的数目。
示例 1:
输入:nums = [-1,1,2,3,1], target = 2 输出:3 解释:总共有 3 个下标对满足题目描述: - (0, 1) ,0 < 1 且 nums[0] + nums[1] = 0 < target - (0, 2) ,0 < 2 且 nums[0] + nums[2] = 1 < target - (0, 4) ,0 < 4 且 nums[0] + nums[4] = 0 < target 注意 (0, 3) 不计入答案因为 nums[0] + nums[3] 不是严格小于 target 。
示例 2:
输入:nums = [-6,2,5,-2,-7,-1,3], target = -2 输出:10 解释:总共有 10 个下标对满足题目描述: - (0, 1) ,0 < 1 且 nums[0] + nums[1] = -4 < target - (0, 3) ,0 < 3 且 nums[0] + nums[3] = -8 < target - (0, 4) ,0 < 4 且 nums[0] + nums[4] = -13 < target - (0, 5) ,0 < 5 且 nums[0] + nums[5] = -7 < target - (0, 6) ,0 < 6 且 nums[0] + nums[6] = -3 < target - (1, 4) ,1 < 4 且 nums[1] + nums[4] = -5 < target - (3, 4) ,3 < 4 且 nums[3] + nums[4] = -9 < target - (3, 5) ,3 < 5 且 nums[3] + nums[5] = -3 < target - (4, 5) ,4 < 5 且 nums[4] + nums[5] = -8 < target - (4, 6) ,4 < 6 且 nums[4] + nums[6] = -4 < target
解题思路
弄明白一件事就好了。 我们要的是 两个数的和s < target,那么要做的事就是 s>=target的时候更新一下right, s<taget的时候记录一下ans并且更新left。
- s >= target:right = right - 1
- s < target :① ans += right - left ②left = left + 1
对①的解释:当s < target时,意味着nums[left] + nums[right] < target
那么 nums[left] 与 区间[left+1, right]的任意一个数组成的和都 < target。区间[left+1, right]的长度就是nums[left] 与 区间[left+1, right]的任意一个数组成的对数,所以 right - (left+1) + 1 = right - left。
class Solution:
def countPairs(self, nums: List[int], target: int) -> int:
nums.sort()
ans = 0
n = len(nums)
left = 0
right = n - 1
while left < right:
s = nums[left] + nums[right]
if s >= target:
right = right - 1
else:
ans += right - left
left = left + 1
return ans
六、Leetcode611. 有效三角形的个数
给定一个包含非负整数的数组 nums
,返回其中可以组成三角形三条边的三元组个数。
示例 1:
输入: nums = [2,2,3,4] 输出: 3 解释:有效的组合是: 2,3,4 (使用第一个 2) 2,3,4 (使用第二个 2) 2,2,3
示例 2:
输入: nums = [4,2,3,4] 输出: 4
解题思路 这个题好难理解,最后看了大佬的题解才弄明白。
想要构成一个三角形,要求是 任意两边之和 > 第三边。
思路一: 直接暴力破解,我只有这个思路0 0,但是有些样例过不去。
def triangleNumber(self, nums: List[int]) -> int:
n = len(nums)
ans = 0
for i in range(n-2):
for j in range(i+1, n-1):
for k in range(j+1, n):
if nums[i] + nums[j] > nums[k] and nums[i] + nums[k] > nums[j] and nums[j] + nums[k] > nums[i]:
ans += 1
return ans
思路二:
首先将这个数组排好序。
然后枚举 第三条边。将问题任意两边之和 > 第三边 --->变为 a + b > c
原理是 如果数组有序(从小到大),那么 如果 a+b > c,肯定满足任意两边之和 > 第三边。
因为 c是在 a、b的后边,所以 c > b > a ----> a+c > b 以及 b+c > a。
小细节:
- 枚举第三条边, 就要保证前面至少有两个数,所以从 下标2这个位置开始遍历。
- if 前两条边的和 < 第三条边,那么移动第一条边的指针,使其+1,原因如下。
nums[left] + nums[right] < c, 那么 nums[left] + nums[left+1] < c, nums[left]与在[left+1, right]这个区间的数相加都 < c,所以 left = left+1。
- 如果 nums[left] + nums[right] > c ,那么 nums[right]与在[left, right-1]这个区间的数相加都 > c,故 ans += right - left。
def triangleNumber(self, nums: List[int]) -> int:
nums.sort()
n = len(nums)
ans = 0
for i in range(2, n):
c = nums[i]
k = 0
v = i - 1
while k < v:
s = nums[k] + nums[v]
if s > c:
ans += v - k
v = v - 1
else:
k = k + 1
return ans
总结
暂时没什么想说的,继续努力吧。