快乐的LeetCode --- 18. 四数之和

前期回顾：

快乐的LeetCode — 1. 两数之和
快乐的LeetCode — 167. 两数之和 II - 输入有序数组
快乐的LeetCode — 15. 三数之和

题目描述：

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

---

解题思路 : 双指针法

1. 指针依次是 p,k,i,j，如果 nums[p] + 3 * nums[p + 1] > target，因为 nums 按升序排列，所以之后的数肯定都大于 target ，直接 break；
2. 如果 nums[p] + 3 * nums[-1] < target，那么当前的 nums[p] 加其余三个数一定小于 target，故 p 直接下一位即可，continue；
3. k 和 p 判断完全一样，只是将 3 变成了 2，target 变成了 target - nums[p]。

同样地，为了避免结果重复，某个指针遇到相同的数需要直接跳过，这与三数之和是一样的。

---

代码：

class Solution:
    def fourSum(self, nums, target):
        nums.sort()
        n = len(nums)
        res = []
        p = 0  # p, k, i, j
        while p < n - 3:  # 文中提到的条件1和条件2，可以直接跳过
            if nums[p] + 3 * nums[p + 1] > target: break
            if nums[p] + 3 * nums[-1] < target:
                while p < n - 4 and nums[p] == nums[p + 1]: p += 1
                p += 1
                continue
            k = p + 1
            while k < n - 2:   # k 和 p 的判断是一样的
                if nums[p] + nums[k] + 2 * nums[k + 1] > target: break
                if nums[p] + nums[k] + 2 * nums[-1] < target: 
                    while k < n - 3 and nums[k] == nums[k + 1]:
                        k += 1
                    k += 1
                    continue
                i = k + 1
                j = n - 1
                new_target = target - nums[p] - nums[k]
                while i < j:
                    if nums[i] + nums[j] > new_target: j -= 1
                    elif nums[i] + nums[j] < new_target: i += 1
                    else:
                        res.append([nums[p],nums[k],nums[i],nums[j]])
                        i += 1
                        j -= 1
                        while i < j and nums[i] == nums[i - 1]: i += 1 # 避免结果重复
                        while i < j and nums[j] == nums[j + 1]: j -= 1 # 避免结果重复
                while k < n - 3 and nums[k] == nums[k + 1]: k += 1
                k += 1
            while p < n - 4 and nums[p] == nums[p + 1]: p += 1
            p += 1
        return res

s = Solution()
nums = [1, 0, -1, 0, -2, 2]
target = 0
print(s.fourSum(nums, target))

---

题目来源：

https://leetcode-cn.com/problems/4sum