Python三數(shù)之和的實(shí)現(xiàn)方式

更新時(shí)間：2022年05月11日 10:30:19 作者：每天收獲一點(diǎn)點(diǎn)

這篇文章主要介紹了Python三數(shù)之和的實(shí)現(xiàn)方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

三數(shù)之和題目描述

給你一個(gè)包含 n 個(gè)整數(shù)的數(shù)組 nums，判斷 nums 中是否存在三個(gè)元素 a，b，c ，

使得 a + b + c = 0 ？請(qǐng)你找出所有滿足條件且不重復(fù)的三元組。

答案中不允許包含重復(fù)的三元組。

示例：

給定數(shù)組 nums = [-1, 0, 1, 2, -1, -4]，

滿足要求的三元組集合為：

[
? [-1, 0, 1],
? [-1, -1, 2]
]

思路

1. 首先將數(shù)組排序，可以利用Python內(nèi)置函數(shù)，也可以利用另外定義排序算法。

2. 應(yīng)用雙指針?biāo)惴?。固定第一個(gè)數(shù),索引為i，遍歷整個(gè)數(shù)組，第一個(gè)數(shù)也是三個(gè)數(shù)中最小的數(shù)，然后在該數(shù)右面設(shè)置左右兩個(gè)指針l和r,l=i+1，r=len(nums)-1,

3. 判斷這三個(gè)索引指向的元素和與0的大小關(guān)系。

和>0,右指針左移一位；和<0，左指針右移一位。

由于要避免重復(fù)的三元組，所以移動(dòng)左右指針的時(shí)候要跳過相鄰的所有相等的nums[i]。

Python3代碼

#導(dǎo)入計(jì)算時(shí)間的包，調(diào)用系統(tǒng)時(shí)間
from time import * 
#初始時(shí)間
t1 = time()
def threeSum(nums):
    nums.sort()
    n = len(nums)
    res = []
    for i in range(n):
        '''如果相鄰的兩個(gè)數(shù)相等，跳過,避免重復(fù)'''
        if i > 0 and nums[i] == nums[i-1]:
            continue 
        l, r = i+1, n-1
        while l < r:
            if nums[i] + nums[l] + nums[r]>0:
                r -= 1
                while nums[r+1] == nums[r]:
                    r -= 1
            elif nums[i] + nums[l] + nums[r]<0:
                l += 1
                while nums[l-1] == nums[l]:
                    l += 1    
            else:
                res.append([nums[i],nums[l],nums[r]])
                l += 1
                r -= 1
                while nums[l] == nums[l - 1]: l += 1
                while nums[r] == nums[r + 1]: r -= 1
    return res
        
if __name__ == '__main__':
    nums = [-1,0,1,2,-1,-4]
    print(threeSum(nums))
#結(jié)束時(shí)間
t2 = time()
#運(yùn)行時(shí)間
run_time = t2 - t1
print(run_time)

運(yùn)行結(jié)果：

[[-1, -1, 2], [-1, 0, 1]]
#運(yùn)行時(shí)間
0.0010113716125488281

以上代碼有一些思想錯(cuò)誤：

遺漏了如果三個(gè)數(shù)全部大于0，則退出循環(huán)，因?yàn)闆]有滿足條件的結(jié)果。

沒有嚴(yán)格判斷每一次的l<r的條件。

修正后的代碼：

from time import * 
t1 = time()
def threeSum(nums):
    nums.sort()
    n = len(nums)
    res = []
    for i in range(n-2):
        if nums[i] > 0:break
        '''如果相鄰的兩個(gè)數(shù)相等，跳過,避免重復(fù)'''
        if i > 0 and nums[i] == nums[i-1]:
            continue 
        l, r = i+1, n-1
        while l < r:
            if nums[i] + nums[l] + nums[r]>0:
                r -= 1
                while l < r and nums[r-1] == nums[r]:
                    r -= 1
            elif nums[i] + nums[l] + nums[r]<0:
                l += 1
                while l < r and nums[l] == nums[l-1]:
                    l += 1    
            else:
                res.append([nums[i],nums[l],nums[r]])
                l += 1
                r -= 1
                while l < r and nums[l] == nums[l - 1]: l += 1
                while l < r and nums[r] == nums[r + 1]: r -= 1
    return res
        
if __name__ == '__main__':
    nums = [-2,-3,0,0,-2]
    print(threeSum(nums))
t2 = time()
run_time = t2 - t1
print(run_time)

結(jié)果：