【前言】
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]