引流:五点钟科技-CSDN博客
题目:
给你一个由
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
你可以按 任意顺序 返回答案 。
【解题思路】
本题虽然是四数之和,但和我们之前做过的所有x数之和都是换汤不换药,思路是一样的,只不过这里可能需要多一个循环,即4个指针共同来完成此题。
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
if len(nums) < 4:
return []
nums.sort()
n = len(nums) - 1
res = []
for i in range(n-2):
if i and nums[i] == nums[i-1]:
continue
for j in range(i+1, n-1):
if j > i+1 and nums[j] == nums[j-1]:
continue
left = j + 1
right = n
while left < right:
if left > j + 1 and nums[left]==nums[left-1]:
left += 1
# 这里需要及时continue掉,否则如果left+1后仍然满足nums[left]==nums[left-1],
# 继续往下执行,仍然会出现重复的答案,例如当nums=[2, 2, 2, 2, 2],下面的right同理。
continue
if right < n and nums[right] == nums[right+1]:
right -= 1
continue
# 一定要先判断重复项,再在此处判断x和target的关系
x = nums[i]+nums[j]+nums[left]+nums[right]
if x == target:
res.append([nums[i], nums[j], nums[left], nums[right]])
left += 1
right -= 1
elif x < target:
left += 1
else:
right -= 1
return res