18. 4Sum
阿新 • • 發佈:2020-10-05
給定一個包含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]
]
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/4sum
3sum多層for
class Solution:def fourSum(self, nums: List[int], target: int) -> List[List[int]]: if len(nums)<4:return [] res=[] nums.sort() n=len(nums) for i in range(n-3): if i!=0 and nums[i]==nums[i-1]: continue for j in range(i+1,n-2):if j!=i+1 and nums[j]==nums[j-1]: continue l=j+1 r=n-1 while l<r: sum=nums[i]+nums[j]+nums[l]+nums[r] if sum<target: l+=1 elif sum>target: r-=1 else: res.append([nums[i],nums[j],nums[l],nums[r]]) l+=1 r-=1 while l<r and nums[l]==nums[l-1]:l+=1 while l<r and nums[r]==nums[r+1]:r-=1 return res