【LeetCode】四数之和

题目

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

---

Python代码

求四数之和可以使用暴力的方法来破解，复杂度为四次for循环，很不推荐。我们这里用到字典进行数组对的储存。

这里读取的方式采用牛客网的读取，第一个数字为目标求和数字，后面为数字集合。这个解法也是网上目前比较优的一种解法。

my_input = '0,1,0,-1,0,1,-1,-2,2'
a = list(map(int, my_input.split(',')))
#a = list(map(int, input().split(',')))
kk = a.pop(0) # 读入序列的第一个数字为target
n = len(a)
a.sort() # 先进行排序
rest = set() # 构造剩余集合
dic = {} # 构造记录两数求和的字典

#for i in range(n - 3):
#    for j in range(i + 1, n - 2):
#        for k in range(j + 1, n - 1):
#            for l in range(k + 1, n):
#                if (kk - a[i] - a[j] - a[k] - a[l]) == 0:
#                    rest.add((a[k[0]], a[k[1]], a[i], a[j]))


for i in range(n - 1):
    for j in range(i + 1, n):
        num = a[i] + a[j] # 先记录两个数字求和的所有情况，记录在字典中
        
        if num in dic:
        	dic[num].append((i, j)) # 若此求和结果已在字典中，则通过append进行添加
            
        else:
            dic[num] = [(i, j)] # 若没有在字典中，则开辟新的位置储存
            
for i in range(2, n - 1): # 再次遍历
    for j in range(i + 1, n):
        res1 = kk - (a[i] + a[j]) # 记录离目标还差多少
        
        if res1 in dic: # 先判断还差的数字是否能由数组中的数字组合而成
            for k in dic[res1]: # 遍历还差多少，那个数字的所有集合
                if k[1] < i: # 若能组合而成，还需判断原本求的两个与新的两个数字不重合（原本大的那个数字要小于现在小的那个数字，则无重合）
                    rest.add((a[k[0]], a[k[1]], a[i], a[j]))

print(rest)
# print(len(rest)) # 最后计算去重后的组合个数