题:给定一个包含n个整数的数组nums和一个目标值target,判断nums中是否存在四个元素,使他们的和为target?
法:转换为两数之和做,建立一个字典,key值是两数之和,value是他们成对的下标,一个key可能对应多组下标,所以value是列表的形式,表中每个元素都是一个下标对。建立完字典后,再去遍历用两个指针i与j暴力遍历nums,每次都去找tmp=target-nums[i]-nums[j],如果tmp在字典中,说明对于下标i与j,还有另外的两个数与他们相加等于target,找到dict1[key]对应的下标对即可。这里注意,为了避免出现重复四元组,我们先将nums排序,并对于i与j下标,判断找到的新下标是否大于j
def fourSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[List[int]]
"""
length=len(nums)
if length<4:return []
dict1={}
nums.sort()
for i in range (length):
for j in range (i+1,length):
tmp=nums[i]+nums[j]
if tmp not in dict1:dict1[tmp]=[(i,j)]
else:
dict1[tmp].append((i,j))
res=set()
for i in range (length-3):
for j in range (i+1,length-2):
tmp=target-nums[i]-nums[j]
if tmp in dict1:
for k in dict1[tmp]:
if k[0]>j:
res.add((nums[i],nums[j],nums[k[0]],nums[k[1]]))
#print(res)
return list(res)