三数和问题

任给数组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。这样就算遍历到了也是以前遍历过的了。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值