1. 程式人生 > 實用技巧 >18. 4Sum

18. 4Sum

給定一個包含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