LeetCode 三数之和（双指针法）（算法练习）

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。

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

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

nums = [-2,0,0,2,2]
rs = []
# 对数列进行递增排序
nums.sort()
# nums = [-1,-1,-1,0,1,2]
# print(nums)
# 设置最小的第一个 指针 k
for k in range(len(nums) - 2):
    # 设置l,r指针
    r = len(nums) - 1  # 指向最大 K 变化一次再次指向最后
    l = k + 1  # 不断指向下一个变换的
    # 最小的指针指向大于1 与指向相同的结果时跳过
    # print('k: %d,l: %d,r: %d' % (k, l, r))
    # print('nums[k]: %d,nums[l]: %d,nums[r]: %d' % (nums[k], nums[l], nums[r]))
    if nums[k] > 0:
        break
    if (k > 0) and (nums[k] == nums[k - 1]):
        continue
    while (k + 1) <= l < r:
        # 处理相同的指针元素
        if (l > k + 1) and (nums[l] == nums[l - 1]):
            l += 1
            continue
        if (r < len(nums) - 1) and (nums[r] == nums[r + 1]):
            r -= 1
            continue
        s = nums[k] + nums[l] + nums[r]

        # print(s)
        # s == 0 说明结果符合 将索引指向元素记录
        if s == 0:
            rs.append([nums[k], nums[l], nums[r]])
            l += 1
            r -= 1
        # s < 0 的情况 说明 l 指针需要向后移一位
        elif s < 0:
            l += 1
        # s>0 的情况说明 r 指针需要往前移一位
        else:
            r -= 1
print(rs)