leetcode 18. 四数之和（python）

【前言】

           python刷leetcode题解答目录索引：https://blog.csdn.net/weixin_40449300/article/details/89470836

           github链接：https://github.com/Teingi/test 

【正文】

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

字典查找法

总思路：把N4拆成2个N2。第一个for循环，先求后2个值可能的取值的所有情况，并且把它们储存在一个字典里，以和作为键。

第二个for，我们遍历前2个值所可能取的各种值，算出和并且检查target - onesum是否在我们的字典里，如果在，就说明我们找到了一个解。其实这种求几数之和的问题，都可以通过这种思路。比如说三数之和，现在我就想到根本不必用指针，算出所有后2个值所加的可能的和，然后用字典存，最后用target-第一个值去检查是否存在这样的键。

class Solution:
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """

        lens = len(nums)
        dic = {}
        #先排序后用集合去重
        res = set()
        nums.sort()
        #先计算前两个数的和
        for i in range(lens-1):
            for p in range(i+1,lens):
                key = nums[i] + nums[p]

                if key not in dic:
                    dic[key] = [(i, p)]
                else:
                    dic[key].append((i, p))

        for i in range(2,lens-1):
             for p in range(i+1, lens):
                    pre = target - nums[i] - nums[p]
                    if pre in dic:
                        for index in dic[pre]:
                         #通过下标判断为合格的后两位数
                            if index[1] < i:
                                res.add((nums[index[0]], nums[index[1]],nums[i], nums[p]))
        return [list(i) for i in res]