给你一个由 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]]
解题思路:思路与三数之和一致,部分去重的位置有点差别,这里给出双指针解法。
class Solution:
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
result = []
nums.sort()
for i in range(len(nums)):
if i > 0 and nums[i-1] == nums[i]:
continue
for j in range(i+1,len(nums)):
if j>i+1 and nums[j] == nums[j-1]:
continue
left = j+1
right = len(nums)-1
while left < right:
e = target - nums[i] - nums[j] - nums[left] -nums[right]
if e > 0:
left += 1
elif e<0:
right -= 1
else:
result.append([nums[i], nums[j], nums[left], nums[right]])
while left<right and nums[left] == nums[left+1]:
left+=1
while left<right and nums[right] == nums[right-1]:
right -= 1
left += 1
right -= 1
return result