n数之和是一系列题目,题设为在一个数组中,找出全部的不重复的n元组,使得这n个数的和为0(或者为target,原理相同)
例如:两数之和
原题目:力扣
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数并返回它们的数组下标。
解法1:使用哈希表来将搜索target-num的时间复杂度降至,整体时间复杂度,空间复杂度。
def twoSum(nums,target):
hash_map={}
for i,num in enumerate(nums):
if target-num in hash_map: #前面是不是有可以满足题意的数字?
return [i,hash_map[target-num]]
target[num]=i
解法2:先对表排序,然后使用双指针,时间复杂度,空间复杂度。虽然时间复杂度提高了,但是这种思想是后续n数之和的基本思想。
def twoSum(nums,target):
i,j=0,len(nums)-1
nums.sort()
while i<j:
if nums[i]+nums[j]>target:
j-=1
elif nums[i]+nums[j]<target:
i+=1
else:
return [i,j]
从两数之和到三数之和:
原题目:力扣
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0
且不重复的三元组。
解法1:如果我们还是考虑哈希表,原本的寻找target-num 就变成了寻找当前元素前面和为-num的组合,时间复杂度为,空间复杂度。
class Solution:
def threeSum(self, nums: List[int]):
hash_nums = {}
result_list = []
nums.sort()
if nums!=[] and nums[0] > 0:
return []
for ti, tnumi in enumerate(nums):
hash_nums[tnumi] = ti #依旧是以空间换时间,让查找值为某个数的索引这一操作复杂度降到O(1)
for i, numi in enumerate(nums):
if i>0 and nums[i]==nums[i-1]: #如果小的数字都大于0,那后面的就更没希望了
continue
for j, numj in enumerate(nums[i + 1:]):
if j>0 and nums[i+1+j]==nums[j+i]:
continue
c_re = hash_nums.get(-(numi + numj))
if c_re is not None and c_re > j + i+1:
result_list.append([numi, numj, -(numi + numj)])
return result_list
解法2:排序+双指针,定住一个坐标动另外两个。
class Solution:
def threeSum(self, nums: List[int]):
n=len(nums)
res=[]
if(not nums or n<3):
return []
nums.sort()
res=[]
for i in range(n):
if(nums[i]>0):
return res #如果小的元素都大于0,大的就更没希望了
if(i>0 and nums[i]==nums[i-1]): #如果索引为i的元素和前一个相同,那就一定会找到相同的解,故跳过
continue
L=i+1
R=n-1
while(L<R): #从这里开始其实是类似于两数之和的方法
if(nums[i]+nums[L]+nums[R]==0):
res.append([nums[i],nums[L],nums[R]])
while(L<R and nums[L]==nums[L+1]):
L=L+1
while(L<R and nums[R]==nums[R-1]):
R=R-1
L=L+1
R=R-1
elif(nums[i]+nums[L]+nums[R]>0):
R=R-1
else:
L=L+1
return res
同理,我们有在四数之和,n数之和都有类似的做法,该类问题即可得到解决,时间复杂度为。