18. 四数之和
给定一个包含 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
自己想了一个思路,
左右开始定位2个元素,然后查找中间元素中等于 target-左-右的所有组合,然后再和左 和 右的数字组合
代码应该没有问题,不过超时了
class Solution:
def twoSum(self, nums, target):
nums.sort()
minnum=-1
result=[]
for i in range(0,len(nums)):
idx=-1
if minnum>0 and i>=minnum:
return result
try:
idx=nums.index((target-nums[i]))
except Exception as e:
idx=-1
if idx>=0 and idx!=i:
if minnum==-1 or idx<minnum:
minnum=idx
print("%s:%s %s:%s"%(i,nums[i],idx,nums[idx]))
result.append([nums[i],nums[idx]])
return result
def fourSum(self, nums, target):
left=0
right=len(nums)-1
nums.sort()
result=[]
for i in range(0,len(nums)-3):
for j in range(len(nums)-1,i+3,-1):
temp=self.twoSum(nums[i+1:j],target-nums[i]-nums[j])
if len(temp)>0:
for aa in temp:
result.append("%s:%s:%s:%s"%(nums[i],aa[0],aa[1],nums[j]))
aa=[]
for i in list(set(result)):
aa.append(list(map(int, i.split(":"))) )
return aa