LeetCode-18

18. 四数之和

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

Example

input
nums = [1, 0, -1, 0, -2, 2]， target = 0。
output
[ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]

Note

答案中不可以包含重复的四元组。

思路

“三数之和”+“最接近的三数之和”

代码如下

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        n = len(nums)
        if not nums or n <4:
            return []
        nums.sort()
        ans = []
        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
                
                max_ans = nums[i] + nums[j] + nums[n-2] + nums[n-1]
                min_ans = nums[i] + nums[j] + nums[j+1] + nums[j+2]
                if max_ans < target:
                    continue
                if min_ans > target:
                    break
                
                l, r = j+1, n-1
                while l < r:
                    res = nums[i] + nums[j] + nums[l] + nums[r]
                    if res == target:
                        ans.append([nums[i], nums[j], nums[l], nums[r]])
                        while l < r and nums[l] == nums[l+1]:
                            l += 1
                        while l < r and nums[r-1] == nums[r]:
                            r -=1
                        l +=1
                        r -=1
                    elif res > target:
                        r -=1
                    else:
                        l +=1
        
        return ans