题目
给你一个由 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]]
提示
- 1 <= nums.length <= 200
- - 1 0 9 10^9 109 <= nums[i] <= 1 0 9 10^9 109
- - 1 0 9 10^9 109<= target <= 1 0 9 10^9 109
思路及算法代码
思路 方法一:排序 + 双指针
最朴素的方法是使用四重循环枚举所有的四元组,然后使用哈希表进行去重操作,得到不包含重复四元组的最终答案。假设数组的长度是 n,则该方法中,枚举的时间复杂度为O( n 4 n^4 n4),去重操作的时间复杂度和空间复杂度也很高,因此需要换一种思路。
为了避免枚举到重复四元组,则需要保证每一重循环枚举到的元素不小于其上一重循环枚举到的元素,且在同一重循环中不能多次枚举到相同的元素。
为了实现上述要求,可以对数组进行排序,并且在循环过程中遵循以下两点:
- 每一种循环枚举到的下标必须大于上一重循环枚举到的下标;
- 同一重循环中,如果当前元素与上一个元素相同,则跳过当前元素。
使用上述方法,可以避免枚举到重复四元组,但是由于仍使用四重循环,时间复杂度仍是 O( n 4 n^4 n4)。注意到数组已经被排序,因此可以使用双指针的方法去掉一重循环。
使用两重循环分别枚举前两个数,然后在两重循环枚举到的数之后使用双指针枚举剩下的两个数。假设两重循环枚举到的前两个数分别位于下标 i 和 j,其中 i<j。初始时,左右指针分别指向下标 j+1 和下标 n−1。每次计算四个数的和,并进行如下操作:
- 如果和等于 target,则将枚举到的四个数加到答案中,然后将左指针右移直到遇到不同的数,将右指针左移直到遇到不同的数;
- 如果和小于 target,则将左指针右移一位;
- 如果和大于 target,则将右指针左移一位。
使用双指针枚举剩下的两个数的时间复杂度是 O(n),因此总时间复杂度是 O( n 3 n^3 n3),低于 O( n 4 n^4 n4)。
具体实现时,还可以进行一些剪枝操作:
- 在确定第一个数之后,如果
nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target
,说明此时剩下的三个数无论取什么值,四数之和一定大于 target,因此退出第一重循环; - 在确定第一个数之后,如果
nums[i]+nums[n−3]+nums[n−2]+nums[n−1]<target
,说明此时剩下的三个数无论取什么值,四数之和一定小于 target,因此第一重循环直接进入下一轮,枚举nums[i+1]; - 在确定前两个数之后,如果
nums[i]+nums[j]+nums[j+1]+nums[j+2]>target
,说明此时剩下的两个数无论取什么值,四数之和一定大于 target,因此退出第二重循环;
在确定前两个数之后,如果nums[i]+nums[j]+nums[n−2]+nums[n−1]<target
,说明此时剩下的两个数无论取什么值,四数之和一定小于 target,因此第二重循环直接进入下一轮,枚举 nums[j+1]。
代码
from typing import List
class Solution:
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
quadruplets = list() # 存储结果的列表
if not nums or len(nums) < 4: # 如果输入列表为空或长度小于4,则直接返回空列表
return quadruplets
nums.sort() # 对输入列表进行排序,方便后续处理
length = len(nums) # 获取列表的长度
for i in range(length - 3): # 第一个指针范围:0 到 length - 4
if i > 0 and nums[i] == nums[i - 1]: # 如果当前数字与前一个相同,跳过,避免重复结果
continue
if nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target: # 如果当前四个数字的和已经大于目标值,结束循环
break
if nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target: # 如果当前数字加上列表中最后三个数字的和小于目标值,继续循环
continue
for j in range(i + 1, length - 2): # 第二个指针范围:i+1 到 length - 3
if j > i + 1 and nums[j] == nums[j - 1]: # 如果当前数字与前一个相同,跳过,避免重复结果
continue
if nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target: # 如果当前四个数字的和已经大于目标值,结束循环
break
if nums[i] + nums[j] + nums[length - 2] + nums[length - 1] < target: # 如果当前数字加上列表中最后两个数字的和小于目标值,继续循环
continue
left, right = j + 1, length - 1 # 定义左右两个指针,分别指向当前数字的下一个和列表末尾
while left < right: # 当左指针小于右指针时,进行循环
total = nums[i] + nums[j] + nums[left] + nums[right] # 计算当前四个数字的和
if total == target: # 如果等于目标值,将结果添加到结果列表中,并调整左右指针
quadruplets.append([nums[i], nums[j], nums[left], nums[right]]) # 将当前四个数字加入结果列表
while left < right and nums[left] == nums[left + 1]: # 跳过左指针重复的数字
left += 1
left += 1 # 左指针右移一位
while left < right and nums[right] == nums[right - 1]: # 跳过右指针重复的数字
right -= 1
right -= 1 # 右指针左移一位
elif total < target: # 如果当前四个数字的和小于目标值,移动左指针
left += 1
else: # 如果当前四个数字的和大于目标值,移动右指针
right -= 1
return quadruplets # 返回结果列表
复杂度分析
时间复杂度: O( n 3 n^3 n3),其中 n 是数组的长度。排序的时间复杂度是 O(nlogn),枚举四元组的时间复杂度是 O( n 3 n^3 n3),因此总时间复杂度为 O( n 3 n^3 n3+nlogn)=O( n 3 n^3 n3)。
空间复杂度:O(logn),其中 n 是数组的长度。空间复杂度主要取决于排序额外使用的空间。此外排序修改了输入数组 nums,实际情况中不一定允许,因此也可以看成使用了一个额外的数组存储了数组 nums 的副本并排序,空间复杂度为 O(n)。