Leetcode18. 四数之和

一、题目描述：

给你一个由 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
• 你可以按 任意顺序 返回答案 。

1. 示例 1：
   • 输入：nums = [1,0,-1,0,-2,2], target = 0
   • 输出：$[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]$
2. 示例 2：
   • 输入：nums = [2,2,2,2,2], target = 8
   • 输出：$[[2,2,2,2]]$

• 提示：
  • $1\le nums.length\le 200$
  • $-109\le nums[i]\le 109$
  • $-109\le target\le 109$

二、解决思路和代码

1. 解决思路

   • 分析：四数之和的解题思路同三数之和 详见 Leetcode15. 三数之和 解题思路分析

2. 代码

   from typing import *
   
   class Solution:
       def quickSort(self, nums: List[int], start: int, end: int):   
           if start>end: return
           left, right = start, end
           temp = nums[left]
           while left < right:
               while left<right and nums[right]>=temp: right-=1
               if left < right:
                   nums[left] = nums[right]
                   left += 1
                   
               while left<right and nums[left]<=temp: left += 1
               if left<right:
                   nums[right] = nums[left]
                   right -= 1
           nums[left] = temp
           self.quickSort(nums, start, left-1)
           self.quickSort(nums, left+1, end)
           return
       
       def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
           if len(nums)<4: return []
           nums2 = nums.copy()
           self.quickSort(nums2, 0, len(nums2)-1)
           res = []
           
           for i in range(len(nums2)-3):
               for j in range(i+1, len(nums2)-2):
                   l, r = j+1, len(nums2)-1
                   while l<r:
                       temp = [nums2[i], nums2[j], nums2[l], nums2[r]]
                       if sum(temp)==target:
                           if temp not in res: res.append(temp)
                           l += 1
                           r -= 1
                       while l<r and sum([nums2[i], nums2[j], nums2[l], nums2[r]])>target: r-=1
                       while l<r and sum([nums2[i], nums2[j], nums2[l], nums2[r]])<target: l+=1
           return res