LeetCode题解(python)-18. 四数之和

LeetCode题解（python）

18. 四数之和

题目描述

给定一个包含 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]
]

解题心得

本题是在解题三数之和（3月13日）的基础上多加一次循环的双指针“夹击”的四数之和的题目。

i和t是最前面用于遍历的两个数，逐渐往后移动，其中t=i+1

j和k是t后面的数列的第一个和最后一个的指针，从两边向中间遍历，直到找到符合题意的四个数字。

注意：要求不可包含重复的四元组

list（set（rel））

解题代码

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        nums.sort()
        numsort = nums
        rel = []
        for i in range(len(numsort)):

            if i!=0 and numsort[i] == numsort[i-1]:
                continue
            for t in range(i+1,len(numsort)):
                if t!=i+1 and numsort[t] == numsort[t-1]:
                    continue
                j = t+1
                k = len(numsort) -1 
                while j<k:
                    if (numsort[j]+numsort[k]+numsort[i]+numsort[t]) == target:
                        rel.append((numsort[i],numsort[t],numsort[j],numsort[k]))
                        j = j+1
                        k = k-1
                    elif (numsort[j]+numsort[k]+numsort[i]+numsort[t]) < target:
                        j = j+1
                    else:
                        k = k-1
        return list(set(rel))