Leetcode018 4sum

四数之和

题目描述：

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

---

解题思路：

解法思路同样遵循低16题“最接近的三数之和”的方法

 整体思路和三数之和相似，因为题目要求为四数之和，可以将前两个数的和看作是一个新的数值，然后将其同后两个数字进行“三数之和的运算”，这样可以类比三数之和的解题方法。

• 两层遍历，取出前两个数，将其存放在数组中
• 从数组中依次取出前两个数，并求其和，看作为一个新的数值
• 将该新数值同之后两个遍历的数组取出的两个值进行“三数之和”，求得答案。

---

Python源码：

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        res, dicti = set(), {}
        numLen = len(nums)
        nums.sort()
        for i in range(numLen):
            for j in range(i + 1, numLen):
                key = nums[i] + nums[j]
                if key not in dicti.keys():
                    dicti[key] = [(i, j)]
                else:
                    dicti[key].append((i, j))
        for i in range(numLen):
            for j in range(i + 1, numLen - 2):
                exp = target - nums[i] - nums[j]
                if exp in dicti.keys():
                    for tmpIndex in dicti[exp]:
                        if tmpIndex[0] > j:
                            res.add((nums[i], nums[j], nums[tmpIndex[0]], nums[tmpIndex[1]]))
        return [list(i) for i in res]

欢迎关注我的github：https://github.com/UESTCYangHR