leetcode 15.三数之和

题目描述

给定一个包含 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

算法流程：

• 特判，对于数组长度 n，如果数组为 null或者数组长度小于 3，返回 []。

1. 对数组进行排序。
2. 遍历排序后数组：
   • 若 nums[i]>0：因为已经排序好，所以后面不可能有三个数加和等于 0，直接返回结果。
   • 对于重复元素：跳过，避免出现重复解
   • 令左指针 L=i+1，右指针 R=n-1，当 L<R 时，执行循环：
     • 当nums[i]+nums[L]+nums[R]==0，执行循环，判断左界和右界是否和下一位置重复，去除重复解。并同时将 L,R移到下一位置，寻找新的解
     • 若和大于 0，说明 nums[R] 太大，R 左移
     • 若和小于 0，说明 nums[L] 太小，L 右移

复杂度分析

时间复杂度：
$$O\left(n^2\right)，数组排序O(N\log N)，遍历数组O\left(n\right)，双指针遍历O\left(n\right)，总体O(NlogN)+O(n)\ast O(n)，O\left(n^2\right)$$

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

思路参考：https://leetcode-cn.com/problems/3sum/solution/pai-xu-shuang-zhi-zhen-zhu-xing-jie-shi-python3-by/

python3

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        n = len(nums)
        if(n<3 or not nums):
            return []
        res = []
        nums.sort()
        for i in range(n):
            #若 nums[i]>0：因为已经排序好，所以后面不可能有三个数加和等于 0，直接返回结果
            if(nums[i]>0):
                return res
            #对于重复元素：跳过，避免出现重复解
            if(i>0 and nums[i]==nums[i-1]):
                continue
            L = i+1
            R = n - 1
            while(L<R):
                if(nums[i]+nums[L]+nums[R]==0):
                    res.append([nums[i],nums[L],nums[R]])
                    while(L<R and nums[L]==nums[L+1]):
                        L+=1
                    while(L<R and nums[R]==nums[R-1]):
                        R-=1
                    L+=1
                    R-=1
                elif(nums[i]+nums[L]+nums[R]>0):
                    R-=1
                else:
                    L+=1
        return res

c++

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n = nums.size();
        if(n == 0){
            return {};
        }
        sort(nums.begin(), nums.end()); 
        vector<vector<int>> res;
        for(int i=0;i<n;i++){
             if(nums[i]>0){
                 return res;
             }
            if(i>0 && nums[i]==nums[i-1])
                continue;
            int L = i+1;
            int R = n-1;
            while(L<R){
                if(nums[i]+nums[L]+nums[R]==0){
                    res.push_back(vector<int>{nums[i],nums[L],nums[R]});
                    while(L<R && nums[L]==nums[L+1])
                        L++;
                    while(L<R && nums[R]==nums[R-1])
                        R--;
                    L++;
                    R--;
                }
                else if(nums[i]+nums[L]+nums[R]>0){
                    R--;
                }
                else{
                    L++;
                }
            }
            
        }
        return res;
    }
};

golang

func threeSum(nums []int) [][]int {
    n := len(nums)
    answer := [][]int{}
    if n<3 {
        return answer
    }
    sort.Ints(nums)
    for i:=0;i<n;i++ {
        if nums[i]>0{
            return answer
        }
        if i>0 && nums[i]==nums[i-1]{
            continue
        }
        L:=i+1
        R:=n-1
        for L<R{
            if nums[i]+nums[L]+nums[R]==0{
                answer=append(answer,[]int{nums[i],nums[L],nums[R]})
                for L<R && nums[L]==nums[L+1]{
                    L++
                }
                for L<R && nums[R]==nums[R-1]{
                    R--
                }
                L++
                R--
            } else if nums[i]+nums[L]+nums[R]>0{
                R--
            }else{
                L++
            }
        }
    }
    return answer
}