数据结构【双指针、数组】| leetcode 15. 三数之和(中等)

代码链接：https://leetcode.cn/problems/3sum/solution/3sumpai-xu-shuang-zhi-zhen-yi-dong-by-jyd/

解题思路：三指针

暴力法搜索为 $O(N^3)$时间复杂度，可通过双指针动态消去无效解来优化效率。

双指针法铺垫： 先将给定 nums 排序，复杂度为 $O(NlogN)$。

双指针法思路： 固定 3 个指针中最左（最小）数字的指针 $k$，双指针 $i，j$ 分设在数组索引 $\text{(k+1, len(nums)-1)}$ 两端，通过双指针交替向中间移动，记录对于每个固定指针 $k$ 的所有满足 $nums[k]+nums[i]+nums[j]==0$ 的 $i,j$ 组合：

• 当 $nums[k]>0$ 时直接break跳出：因为 $nums[j]\ge nums[i]\ge nums[k]>0$，即 3 个数字都大于 0，在此固定指针 $k$ 之后不可能再找到结果了。
• 当 $k>0$且$nums[k]==nums[k-1]$时即跳过此元素$nums[k]$：因为已经将 $nums[k-1]$ 的所有组合加入到结果中，本次双指针搜索只会得到重复组合。

$i,j$分设在数组索引 $(k+1,len(nums)-1)$ 两端，当 $i<j$ 时循环计算$total=nums[k]+nums[i]+nums[j]$，并按照以下规则执行双指针移动：

• 当$total<0$时，$i+=1$；（不需要跳过重复数字，下一轮循环条件不会满足，自动跳出）
• 当$total>0$时，$j-=1$；
• 当$total==0$时，记录组合$[k,i,j]$至$res$，执行 $i+=1$ 和 $j-=1$ 并跳过所有重复的$nums[i]$和$nums[j]$，防止记录到重复组合。

复杂度分析：

• 时间复杂度 $O(N^2)$：其中固定指针 $k$ 循环复杂度 $O(N)$，双指针 $i,j$ 复杂度 $O(N)$。
• 空间复杂度 $O(1)$：指针使用常数大小的额外空间。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()   # 对数组进行排序
        res = []
        for k in range(len(nums)-2):  # 固定一个数
            # nums[j]≥nums[i]≥nums[k]>0，即3个数字都大于0
            if nums[k] > 0:
                break
            # 跳过此元素nums[k]：因为已经将nums[k - 1]的所有组合加入到结果中
            if k > 0 and nums[k] == nums[k-1]:
                continue

            # k每遍历一次数组，i和j从k+1 -> len(nums)-1移动一次
            i, j = k+1, len(nums)-1
            while i < j:
                total = nums[k] + nums[i] + nums[j] # 求三个数的和
                if total > 0:   # 值大于0, j前移
                    j -= 1
                elif total < 0: # 值小于0, i后移
                    i += 1
                else:   # 值=0
                    res.append([nums[k], nums[i], nums[j]])
                    # 如果临近数值相同，则i指针后移
                    while i < j and nums[i] == nums[i+1]:
                        i += 1
                    # 如果临近数值相同，则j指针前移
                    while i < j and nums[j] == nums[j-1]:
                        j -= 1
                    i += 1  # 移动指针,查找其它可能结果
                    j -= 1
        return res