前期回顾:
快乐的LeetCode — 1. 两数之和
快乐的LeetCode — 167. 两数之和 II - 输入有序数组
快乐的LeetCode — 15. 三数之和
题目描述:
给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d
的值与 target 相等?找出所有满足条件且不重复的四元组。
注意:答案中不可以包含重复的四元组。
示例:
给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。
满足要求的四元组集合为:
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]
解题思路 : 双指针法
- 指针依次是
p,k,i,j
,如果nums[p] + 3 * nums[p + 1] > target
,因为nums
按升序排列,所以之后的数肯定都大于target
,直接break
; - 如果
nums[p] + 3 * nums[-1] < target
,那么当前的nums[p]
加其余三个数一定小于target
,故p
直接下一位即可,continue
; - k 和 p 判断完全一样,只是将 3 变成了 2,target 变成了
target - nums[p]
。
同样地,为了避免结果重复,某个指针遇到相同的数需要直接跳过,这与三数之和是一样的。
代码:
class Solution:
def fourSum(self, nums, target):
nums.sort()
n = len(nums)
res = []
p = 0 # p, k, i, j
while p < n - 3: # 文中提到的条件1和条件2,可以直接跳过
if nums[p] + 3 * nums[p + 1] > target: break
if nums[p] + 3 * nums[-1] < target:
while p < n - 4 and nums[p] == nums[p + 1]: p += 1
p += 1
continue
k = p + 1
while k < n - 2: # k 和 p 的判断是一样的
if nums[p] + nums[k] + 2 * nums[k + 1] > target: break
if nums[p] + nums[k] + 2 * nums[-1] < target:
while k < n - 3 and nums[k] == nums[k + 1]:
k += 1
k += 1
continue
i = k + 1
j = n - 1
new_target = target - nums[p] - nums[k]
while i < j:
if nums[i] + nums[j] > new_target: j -= 1
elif nums[i] + nums[j] < new_target: i += 1
else:
res.append([nums[p],nums[k],nums[i],nums[j]])
i += 1
j -= 1
while i < j and nums[i] == nums[i - 1]: i += 1 # 避免结果重复
while i < j and nums[j] == nums[j + 1]: j -= 1 # 避免结果重复
while k < n - 3 and nums[k] == nums[k + 1]: k += 1
k += 1
while p < n - 4 and nums[p] == nums[p + 1]: p += 1
p += 1
return res
s = Solution()
nums = [1, 0, -1, 0, -2, 2]
target = 0
print(s.fourSum(nums, target))
题目来源: