【python-leetcode15-雙指針】三個數之和為零

問題描述：

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

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

示例：

給定數組 nums = [-1, 0, 1, 2, -1, -4]，

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

代碼：

class Solution:
    def threeSum(self, nums):
        #數組長度
        n = len(nums)
        if (not nums or n < 3):
            return []
        #先將數組進行排序
        nums.sort()
        #保存結果
        res = []
        #從左開始遍歷數組
        for i in range(n):
            #當遍歷到正數時就可以返回結果了
            if (nums[i] > 0):
                return res
            #如果i>0且相鄰兩個值相等，則繼續
            if (i > 0 and nums[i] == nums[i - 1]):
                continue
            #左指針指向i的下一位
            L = i + 1
            #右指針指向數組右端
            R = n - 1
            #循環條件
            #核心就是在第i位時，考慮從i+1位到末尾，不斷通過增加左指針指向的值大小
            #和減少右指針指向的值的大小來找到一個平衡位置使三者之和為0
            while (L < R):
                #如果這三個數加起來為0
                if (nums[i] + nums[L] + nums[R] == 0):
                    #加入結果
                    res.append([nums[i], nums[L], nums[R]])
                    #此時對左指針的下一位進行判斷，如果和其相同，左指針繼續右移
                    while (L < R and nums[L] == nums[L + 1]):
                        L = L + 1
                    #同理，右指針繼續左移
                    while (L < R and nums[R] == nums[R - 1]):
                        R = R - 1
                    #如果不相同直接左指針右移一位
                    L = L + 1
                    #如果不相同直接右指針左移一位
                    R = R - 1
                elif (nums[i] + nums[L] + nums[R] > 0):
                    #如果大於0，右指針左移一位
                    R = R - 1
                else:
                    #如果小於0，左指針右移一位
                    L = L + 1
        return res

結果：