四数之和
题目描述:
给定一个包含 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