[LeetCode-Python版]相向双指针——167. 两数之和 II - 输入有序数组 & 15. 三数之和

167. 两数之和 II - 输入有序数组

题目

给你一个下标从 1 开始的整数数组 numbers ，该数组已按 非递减顺序排列 ，请你从数组中找出满足相加之和等于目标数 target 的两个数。如果设这两个数分别是 numbers[index1] 和 numbers[index2] ，则 1 <= index1 < index2 <= numbers.length 。

以长度为 2 的整数数组 [index1, index2] 的形式返回这两个整数的下标 index1 和 index2。

你可以假设每个输入 只对应唯一的答案 ，而且你 不可以 重复使用相同的元素。

你所设计的解决方案必须只使用常量级的额外空间。

示例 1：
输入：numbers = [2,7,11,15], target = 9
输出：[1,2]
解释：2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。

示例 2：
输入：numbers = [2,3,4], target = 6
输出：[1,3]
解释：2 与 4 之和等于目标数 6 。因此 index1 = 1, index2 = 3 。返回 [1, 3] 。

示例 3：
输入：numbers = [-1,0], target = -1
输出：[1,2]
解释：-1 与 0 之和等于目标数 -1 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。

提示：

• 2 <= numbers.length <= 3 * 104
• -1000 <= numbers[i] <= 1000
• numbers 按 非递减顺序 排列
• -1000 <= target <= 1000
• 仅存在一个有效答案

题目链接

思路

用相向双指针（l和r分别从两端向中间走）来做：

1. 初始化： 双指针 l , r 分别指向数组 numbers 的左右两端 。
2. 循环，每次计算 s=numbers[l]+numbers[r] 。
   • 若 s>target ，则说明 r 和数组中剩下所有的数相加都比target大，因此指针 r 向左移动减小 r 的值，即 r=r−1 。
   • 若 s<target ，则说明 l 和数组中剩下所有的数相加都比target小，因此指针 l 向右移动增加 l 的值，即 l=l+1 。
   • 若 s=target ，由于题目要求索引从 1 开始，因此返回数组 [l+1,r+1] 。
     若循环结束，则返回空数组，代表无和为 target 的数字组合。

复杂度分析

时间复杂度： $O(n)$
空间复杂度：$O(1)$

参考代码

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        l = 0
        r = len(numbers)-1

        while l<r:
            if numbers[l] + numbers[r] == target:
                return [l+1,r+1]# 下标从1开始
            elif numbers[l] + numbers[r] > target:
                r-=1
            else:
                l+=1
        return []    

15. 三数之和

题目

给你一个整数数组 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 。

提示：

• 3 <= nums.length <= 3000
• $-10^5<=nums[i]<=10^5$

思路

把三数之和转化为两数之和

复杂度分析

时间复杂度： $O(n^2)$
空间复杂度：$O(1)$

参考代码

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        # 0. 这道题的数组是未排序的，所以要先排序
        nums.sort()
        res = []
        #1. 把三数之和的问题转化成两数之和的问题
        # 遍历一遍数组，循环内剩下的两数之和等于外循环当前的值的负数

        for i in range(len(nums)-2): # 因为极端情况是 i，j，k是最后三个数，所以i最多到len(nums)-3
            # 去重，当i和前一个数相等时，会找到一样的组合，此时会重复
            if i>0 and nums[i]==nums[i-1]:
                continue
            # 优化：当此时的nums[i]和最小的两个数加起来大于0时说明不可能有结果，直接break
            if nums[i] + nums[i+1] + nums[i+2] >0:
                break
            # 优化，当此时的nums[i]和最大的两个数加起来小于0时说明nums[i]太小，直接continue
            if nums[i] + nums[-1] + nums[-2] <0:
                continue
            j = i+1
            k = len(nums)-1
            while j < k:
                s = nums[i]+nums[j]+nums[k]
                if s < 0:
                    j+=1
                elif s>0:
                    k-=1
                else:
                    res.append([nums[i],nums[j],nums[k]])
                    
                    j+=1
                    # 去重
                    while j<k and nums[j]==nums[j-1]:
                        j+=1
                    k-=1
                    # 去重
                    while j<k and nums[k]==nums[k+1]:
                        k-=1
        return res