18. 四数之和-CSDN博客

本文链接：https://blog.csdn.net/qq_36583400/article/details/135650148

本文介绍了解决给定整数数组nums和目标值target的四数之和问题的算法，利用排序和四个指针（a,b,c,d）遍历数组，找到满足nums[a]+nums[b]+nums[c]+nums[d]=target且不重复的四元组。

摘要由CSDN通过智能技术生成

题目：

给你一个由 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

你可以按 任意顺序 返回答案。

【解题思路】

本题虽然是四数之和，但和我们之前做过的所有x数之和都是换汤不换药，思路是一样的，只不过这里可能需要多一个循环，即4个指针共同来完成此题。

    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        if len(nums) < 4:
            return []
        nums.sort()
        n = len(nums) - 1
        res = []
        for i in range(n-2):
            if i and nums[i] == nums[i-1]:
                continue
            for j in range(i+1, n-1):
                if j > i+1 and nums[j] == nums[j-1]:
                    continue
                left = j + 1
                right = n
                while left < right:
                    if left > j + 1 and nums[left]==nums[left-1]:
                        left += 1
                        # 这里需要及时continue掉，否则如果left+1后仍然满足nums[left]==nums[left-1]，
                        # 继续往下执行，仍然会出现重复的答案，例如当nums=[2, 2, 2, 2, 2]，下面的right同理。
                        continue
                    if right < n and nums[right] == nums[right+1]:
                        right -= 1
                        continue
                    # 一定要先判断重复项，再在此处判断x和target的关系
                    x = nums[i]+nums[j]+nums[left]+nums[right]
                    if x == target:
                        res.append([nums[i], nums[j], nums[left], nums[right]])
                        left += 1
                        right -= 1
                    elif x < target:
                        left += 1
                    else:
                        right -= 1
        return res