python的双指针法：计算整数列表中符合a+b+c=0或者a+b+c+d=0的所有组合

假定给定已排序的列表 nums = [-5, -3, -3, -2, -1, -1, 0, 0, 1, 1, 2, 2, 4, 5, 5, 5, 6, 7, 7]，这里我们需要找出所有符合 a+b+c=0 的不重复组合，比如：

[-5, -1, 6], [-1, 0, 1] .......

这里我们使用双指针法找出所有的组合（自行了解双指针，简单来说就是左边的数往右增大移动，右边的数往左减小移动）。

直接给出 a+b+c=0 的代码：

nums = [-5, -3, -3, -2, -1, -1, 0, 0, 1, 1, 2, 2, 4, 5, 5, 5, 6, 7, 7]
# 这里我们转换成找出 b + c = -a 的过程， 且 a < b < c
# 假定 a 是固定移动的，枚举出 a
for a in range(len(nums)):
    # 如果 a>0, a+b+c>0 退出循环
    if nums[a] > 0:
        break
    # 比如 a 枚举到 -3 的时候，只需要枚举第一个 -3 即可，第二个跳过
    if a > 0 and nums[a] == nums[a-1]:
        continue

    left = a + 1     # b 的索引，左指针
    right = len(nums) - 1  # c 的索引，右指针
    
    # b < c 为条件
    while right > left:
        # 满足 a+b+c=0，打印出来
        if nums[a] + nums[left] + nums[right] == 0:
            print([nums[a], nums[left], nums[right]])

            # 满足 a+b+c=0 条件下，进行下一次查找
            # 比如 a=-2 b=-1(第一次遇到) c=1，则左指针往右移动的过程中下一个还是 -1，不需要重复统计
            while left < right and nums[left] == nums[left+1]:
                left += 1
            # 比如 a=-2 b=-1 c=1(第一次遇到)，则右指针往左移动的过程中下一个还是 1，不需要重复统计
            while left < right and nums[right] == nums[right-1]:
                right -= 1
            
            # 更新下一次判断需要的指针
            left += 1
            right -= 1
        # a 是固定的，因为 b 不能减小，a+b+c>0 意味着 c 要减小
        elif nums[a] + nums[left] + nums[right] > 0:
            right -= 1
        # a 是固定的，因为 c 不能增大，a+b+c<0 意味着  b 要增大
        elif nums[a] + nums[left] + nums[right] < 0:
            left += 1

输出：

[-5, -2, 7]
[-5, -1, 6]
[-5, 0, 5]
[-5, 1, 4]
[-3, -3, 6]
[-3, -2, 5]
[-3, -1, 4]
[-3, 1, 2]
[-2, 0, 2]
[-2, 1, 1]
[-1, -1, 2]
[-1, 0, 1]

这里给出 a+b+c+d=0 的代码：

nums = [-5, -3, -3, -2, -1, -1, 0, 0, 1, 1, 2, 2, 4, 5, 5, 5, 6, 7, 7]

for i in range(len(nums)-3):
    if i > 0 and nums[i] == nums[i-1]:
        continue

    for j in range(i+1, len(nums)-2):
        if j > i+1 and nums[j] == nums[j-1]:
            continue
        
        left = j + 1
        right = len(nums) - 1

        while right > left:
            if nums[i] + nums[j] + nums[left] + nums[right] == 0:
                print([nums[i], nums[j], nums[left], nums[right]])

                while right > left and nums[left] == nums[left+1]:
                    left += 1
                while right > left and nums[right] == nums[right-1]:
                    right -= 1

                left += 1
                right -= 1
            elif nums[i] + nums[j] + nums[left] + nums[right] > 0:
                right -= 1
            elif nums[i] + nums[j] + nums[left] + nums[right] < 0:
                left += 1
                    

输出：

[-5, -3, 1, 7]
[-5, -3, 2, 6]
[-5, -2, 0, 7]
[-5, -2, 1, 6]
[-5, -2, 2, 5]
[-5, -1, -1, 7]
[-5, -1, 0, 6]
[-5, -1, 1, 5]
[-5, -1, 2, 4]
[-5, 0, 0, 5]
[-5, 0, 1, 4]
[-5, 1, 2, 2]
[-3, -3, -1, 7]
[-3, -3, 0, 6]
[-3, -3, 1, 5]
[-3, -3, 2, 4]
[-3, -2, -1, 6]
[-3, -2, 0, 5]
[-3, -2, 1, 4]
[-3, -1, -1, 5]
[-3, -1, 0, 4]
[-3, -1, 2, 2]
[-3, 0, 1, 2]
[-2, -1, -1, 4]
[-2, -1, 1, 2]
[-2, 0, 0, 2]
[-2, 0, 1, 1]
[-1, -1, 0, 2]
[-1, -1, 1, 1]
[-1, 0, 0, 1]