1. 程式人生 > >leetcode15--三數之和

leetcode15--三數之和

題目

給定一個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。

注意:答案中不可以包含重複的三元組。

例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4],

滿足要求的三元組集合為:
[
[-1, 0, 1],
[-1, -1, 2]
]

思路

給定一個數列a+b+c=0,求的問題可以轉換為求a+b=-c:, 我們將輸入的list排序後,給定nums[i], 我們需要尋找 nums[j]+nums[k] = - nums[i], 這裡的 。
題目要求不可包含重複的解,對於排序後的陣列,如果num[i] = num[i+1],那麼他們獲得的nums[j]和nums[k]肯定是相同的(如果解存在的話),所以應該跳過這種情況。同理,如果num[j] = num[j+1], 那麼也會找到相同的nums[k]。這種情況也應該跳過。

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        #儲存結果列表
        res=[]
        #對nums列表進行排序,無返回值,排序直接改變nums順序
        nums.sort()
        # 對排序後的nums進行遍歷
        for i in range(len(nums)):
            #如果排序後第一個數都大於0,則跳出迴圈,不可能有為0 的三數之和
if nums[0]>1: break # 排序後相鄰兩數如果相等,則跳出當前迴圈繼續下一次迴圈,相同的數只需要計算一次 if i>0 and nums[i]==nums[i-1]: continue #記錄i的下一個位置 j=i+1 #最後一個元素的位置 k=len(nums)-1 while j<
k: #判斷三數之和是否為零 if nums[j]+nums[k] ==-nums[i]: #把結果加入陣列之中 res.append([nums[i],nums[j],nums[k]]) #判斷j相鄰元素是否相等,有的話直接跳過這個 while j<k and nums[j] ==nums[j+1]: j+=1 while j<k and nums[k]==nums[k-1]: k-=1 #沒有相等則j+1,k-1,縮小範圍 j+=1 k-=1 #小於-nums[i]的話還能繼續往後取 elif nums[j]+ nums[k] < -nums[i] : j +=1 else: k-=1 return res