18. 四数之和

题目

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

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2：

输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

提示：

• 1 <= nums.length <= 200
• -109 <= nums[i] <= 109
• -109 <= target <= 109

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

1. 双指针

执行用时：44 ms, 在所有 Python 提交中击败了95.69%的用户
内存消耗：12.9 MB, 在所有 Python 提交中击败了85.78%的用户
通过测试用例：293 / 293

class Solution(object):
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        li = list()
        # 如果nums为空或者不满足4个数，返回空
        if not nums or len(nums) < 4:
            return li
        # 排序方便遍历
        nums.sort()
        n = len(nums)
        # 遍历到n-3就可结束
        for i in range(n - 3):
            # 如果当前i值和上一次便利的i值一样，直接跳过
            if i > 0 and nums[i] == nums[i - 1]:
                continue
            # 如果前四数的和比target大，说明接下来都没必要遍历了
            if sum(nums[i:i+4]) > target:
                break
            # 如果当前值加上后三位的数都比target小，那么遍历下一个i
            if nums[i] + sum(nums[-3:]) < target:
                continue
            # 对于第二个数，起始位置为i+1，同时不大于n-2
            for j in range(i + 1, n - 2):
                # 判断和第一个数一样的情况
                if j > i + 1 and nums[j] == nums[j - 1]:
                    continue
                if nums[i] + sum(nums[j:j+3]) > target:
                    break
                if nums[i] + nums[j] + sum(nums[-2:]) < target:
                    continue
                # 最后设置双指针
                l, r = j + 1, n - 1
                while l < r:
                    # 四数和
                    result = nums[i] + nums[j] + nums[l] + nums[r]
                    # 得到想要的结果
                    if result == target:
                        li.append([nums[i], nums[j], nums[l], nums[r]])
                        # 如果还有剩余的数未被遍历，使用2个whileloop跳过相同数
                        while l < r and nums[l] == nums[l + 1]:
                            l += 1
                        l += 1
                        while l < r and nums[r] == nums[r - 1]:
                            r -= 1
                        r -= 1
                    # 如果值小于target，左指针右移
                    elif result < target:
                        l += 1
                    # 大于则右指针左移
                    else:
                        r -= 1
        return li