Leecode15. 三数之和——Leecode大厂热题100道系列

Posted 2021-10-27 来老铁干了这碗代码

我是小张同学，立志用最简洁的代码做最高效的表达

---

以下是我个人做的题解，每个题都尽量囊括了所有解法，并做到了最优解，欢迎大家收藏！留言！

传送门——>Leecode大厂热题100道系列题解

---

问题描述

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

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

示例 1：
输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

示例 2：
输入：nums = []
输出：[]

示例 3：
输入：nums = [0]
输出：[]

提示：
0 <= nums.length <= 3000
-105 <= nums[i] <= 105

---

核心思路

排序 + 双指针 + 固定消元法

做法

1. 特判，对于数组长度 $n$，如果数组为 $null$ 或者数组长度小于 $3$，返回 $[]$。
2. 对数组进行排序。
3. 遍历排序后数组：
   1. 若 $nums[i]>0$：因为已经排序好，所以后面不可能有三个数加和等于 $0$，直接返回结果。
   2. 对于重复元素：跳过，避免出现重复解
   3. 令左指针 $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$ 右移

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        set<vector<int>> s;
        sort(nums.begin(), nums.end());
        for(int i = 0; i < nums.size(); i++) {
            // 去重
            if(i > 0 && nums[i] == nums[i-1]) continue;
            // 如果 > 0 代表后面的都大于0
            if(nums[i] > 0) break;
            int l = i + 1, r = nums.size()-1;
            while(l < r) {
                int t = nums[i] + nums[l] + nums[r];
                if(t == 0) {
                    res.push_back({nums[i], nums[l], nums[r]});
                    while(l < r && nums[l] == nums[l+1]) l++;
                    while(l < r && nums[r] == nums[r-1]) r--;
                    r--; l++;
                } else if(t > 0) {
                    r--;
                } else if(t < 0) {
                    l++;
                }

            }
        }
        return res;
    }
};