Python|双指针解决三数之和问题

def threeSum(nums):

        '''

        算法思路：最外层控制一个元素的循环，内层用双指针，一个从头到尾扫描，另一个从尾到头扫描，判断三个元素的值之和是否为零       

        注意：相同的元素需要跳过

        '''

        nums.sort()# 对列表进行排序

        res = []

        k=0

        for k in range(len(nums) - 2):

            # 如果出现最小元素为正数，则不存在和为0的情况，直接返回

            if nums[k] > 0:

                break

            # 如果出现第一个元素重复的情况，为避免重复结果，跳过后续执行

            if k > 0 and nums[k] == nums[k - 1]:

                continue

            # 定义接下来的两个元素的双指针

            i, j = k + 1, len(nums) - 1

            while i < j:

                s = nums[k] + nums[i] + nums[j]

                if s < 0:

                    i += 1

                    # 跳过重复元素

                    while i < j and nums[i] == nums[i - 1]:

                        i += 1

                elif s > 0:

                    j -= 1

                    # 跳过重复元素

                    while i < j and nums[j] == nums[j + 1]:

                        j -= 1

                else:

                    # 当出现元素满足条件时，将结果加入到列表

                    res.append([nums[k], nums[i], nums[j]])

                    # 接着更新索引（注意跳过相同元素）

                    i += 1

                    j -= 1

                    while i < j and nums[i] == nums[i - 1]:

                        i += 1

                    while i < j and nums[j] == nums[j + 1]:

                        j -= 1

        return res