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

更新時間：2022年05月11日 10:30:19 作者：每天收獲一點點

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

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

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

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

答案中不允許包含重復的三元組。

示例：

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

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

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

思路

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

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

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

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

由于要避免重復的三元組，所以移動左右指針的時候要跳過相鄰的所有相等的nums[i]。

Python3代碼

#導入計算時間的包，調(diào)用系統(tǒng)時間
from time import * 
#初始時間
t1 = time()
def threeSum(nums):
    nums.sort()
    n = len(nums)
    res = []
    for i in range(n):
        '''如果相鄰的兩個數(shù)相等，跳過,避免重復'''
        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é)束時間
t2 = time()
#運行時間
run_time = t2 - t1
print(run_time)

運行結(jié)果：

[[-1, -1, 2], [-1, 0, 1]]
#運行時間
0.0010113716125488281

以上代碼有一些思想錯誤：

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

沒有嚴格判斷每一次的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
        '''如果相鄰的兩個數(shù)相等，跳過,避免重復'''
        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é)果：