任给数组nums=[4,1.5,7,4,4,3],target=12
target为数组中任意的三个元素之和
输出一个数result表示数组中有多少组数和为target
此处:1+3+5=12
4+4+4=12
所以 result=2
对于这道题我们会有两个思路:第一个通过循环设置三个指针,来分别确定三个数,暴力算法去把所有的三个数的组合枚举出来,当这个组合的合等于target时便可以让result+1
def func(nums,target):
result=0
for i in range(len(nums)-2):
for j in range(i,len(nums)-1):
for m in range(j,len(nums-2)):
if nums[i]+nums[j]+nums[m]==target:
result+=1
return result
此时时间复杂度达到了o(n**3)这个时间复杂度是很高的,因此我们想一想能不能进一步优化,此时我们可以参考两数之和的问题。
target=a+b+c
我们可以将其变为target-a=b+c
也就是说问题变为了在除去当前元素a的数组中找到两个元素和为target-a问题也就变为了两数之和问题,此时我们当然也可以继续通过循环遍历去暴力解,但时间复杂度上将会和第一个想法一样没有得到减少,因此我们可以想一想如何两数之和其他的解题方法。此时不难想到通过双指针去解题。
而在两数之和中双指针的使用需要其是有序数组,因此我们在使用双指针之前需要先对数组进行排序。
def(nums,target):
result=0
nums=sorted(nums)
n=len(nums)
for i in range(n-2):
target1=target-num[i]
per=i+1
behind=n-1
while per<behind:
if nums[per]+nums[behind]>target1:
behind-=1
elif nums[per]+nums[behind]<target1:
per+=1
else:
result+=1
break
return result
此处可能大家会有点疑惑,为什么我们减去当前元素之后,进行两数之和的操作是不需要进行对当前元素,之前的元素进行遍历。
此处原因是因为我们在预处理阶段就对nums数组进行了升序处理
也就是说target=a+b+c当我们顺序选择的时候默认了(a>=b>=c)
如果再对当前元素之前的元素进行遍历的话也就是把a当成了b。这样就算遍历到了也是以前遍历过的了。