4. 三数之和_Python算法

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。
示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。
示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

#解题
# 定义解题方法

def fun1(nums):
    nums = list(nums)
    nums.sort()  # 先从小到大排序，再计算可减少大量无效的循环判断
    n = len(nums)
    numsnew = []  # 题目要求返回一个数组
    if (not nums or n < 3):  # 异常处理
        return []
    for i in range(n-2):  # 从第一个开始，并确保后面还有2个数
        if nums[i] > 0:  # 最小大于0和肯定大于不再满足，直接返回
            return numsnew
        if (i > 0 and nums[i] == nums[i - 1]):#默认取第一个之前，往后则先去重，只保留最后一个i
            # print(nums[i - 1])
            continue
        left = i + 1 #初始化左指针
        right = n - 1#初始化右指针
        while left < right:
            if nums[i] + nums[left] + nums[right] == 0:#满足题目条件则加入列表中
                numsnew.append([nums[i], nums[left], nums[right]])
                while left < right and nums[left + 1] == nums[left]:#左边数相同时则移动左指针
                    left = left + 1
                while left < right and nums[right - 1] == nums[right]:#右边数相同时则移动右指针
                    right = right - 1
                left = left + 1 #重置为下一个数标且跟前数不同
                right = right - 1 #重置为下一个数标且跟前数不同
            elif nums[i] + nums[left] + nums[right] > 0:#和大于0移动右指针促使变小继续判断
                right = right - 1
            else: #和小于0移动左指针促使变大继续判断
                left = left + 1

    return numsnew

# 根据题目输入例子进行验证
nums = [-1, 0, 1, 2, -1, -4]

print(fun1(nums))