18 四数之和
给你一个由 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
双指针法,重点还是在剪枝和去重,两层for循环,在三数之和的基础上外面再加一个k,不太一样的是本题target相比三数之和不再是固定的0,所以剪枝条件复杂了一些。在i层剪枝的时候,考虑的是nums[k]+nums[i]和的情况。其他和三数之和一样的逻辑。
class Solution:
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
nums.sort()
res = []
for k in range(len(nums)):
if nums[k] > target and target > 0 and nums[k] > 0:
break
if k > 0 and nums[k] == nums[k-1]:
continue
for i in range(k+1,len(nums)):
if nums[k]+nums[i] > target and target > 0 and nums[k] + nums[i] > 0:
break
if i > k+1 and nums[i] == nums[i-1]:
continue
left, right = i+1, len(nums)-1
while left < right:
if nums[k] + nums[i] + nums[left] + nums[right] == target:
res.append([nums[k], nums[i], nums[left], nums[right]])
while left < right - 1 and nums[left] == nums[left + 1]:
left += 1
while left < right - 1 and nums[right] == nums[right - 1]:
right -= 1
left += 1
right -= 1
if nums[k] + nums[i] + nums[left] + nums[right] < target:
left += 1
if nums[k] + nums[i] + nums[left] + nums[right] > target:
right -= 1
return res