给定一个包含 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]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
思路:无论是几个数的和,本质上都是控制部分,遍历剩下的部分来达到目的。
class Solution(object):
def fourSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[List[int]]
"""
nums.sort()
ans=set()
for i1 in range(len(nums)-3):
for i2 in range(i1+1,len(nums)-2) :
left=i2+1
right=len(nums)-1
while(left<right):
sumnum=nums[i1]+nums[i2]+nums[left]+nums[right]
if(sumnum==target):
ans.add((nums[i1],nums[i2],nums[left],nums[right]))
left+=1
right-=1
elif(sumnum>target):right-=1
else:left+=1
res=[]
for listno in ans:
res.append(list(listno))
return res
小知识:
(1)通过设立set,再把set转为list 的做法可以去掉重复值;
(2)set的添加使用的函数是add,而list使用的是append。