15. 三数之和

15. 三数之和

题目描述

给你一个包含 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\le nums.length\le 3000$
• $-10^5\le nums[i]\le 10^5$

---

题解：

排序 + 双指针。

首先对 nums 排序，这个可以方便我们后面的操作。

对于三元组(i, j, k) ，我们固定 i ， j 和 k 分别从 i + 1 和 n - 1 开始执行以下操作：

• 设 sum = nums[i] + nums[j] + nums[k]

• 若 sum > 0，说明 nums[k] 值太大了，需要变小，于是 --k；

• 若 sum < 0，说明 nums[j] 值太小了，需要变大，于是 ++j；

• 若 sum == 0 ，此时我们找到了一组解，保存到结果中，指针同时移动++j, --k，继续查找结果（需要判重）

判重以测试样例 [-2,-2,0,0,2,2] 为例：

• 在 i = 0, j = 2, k = 5 时找到了一组解，此时 i 固定不动

• 指针移动后 j = 3, k = 4 ，我们可以发现 nums[j] == nums[j - 1], nums[k] == nums[k + 1] ，这些相同元素不跳过，肯定会出现重复结果

• 同理，对于 i 指针，nums[1] == nums[0] ，如果不跳过，同样会出现重复解

同时，在 nums[i] > 0 时直接停止循环，应该能加速不少。

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

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

代码：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n = nums.size();
        if ( n < 3 ) return {};
        sort( nums.begin(), nums.end() );
        int t;
        vector<vector<int>> ret;
        for ( int i = 0; i < n - 2; ++i ) {
            if ( nums[i] > 0 ) break;
            if ( i && nums[i] == nums[i - 1] ) continue;
            for ( int j = i + 1, k = n - 1; j < k; ) {
                t = nums[i] + nums[j] + nums[k];
                if ( t > 0 ) --k;
                else if ( t < 0 ) ++j;
                else {
                    ret.push_back( {nums[i], nums[j++], nums[k--]} );
                    while ( j < k && nums[j] == nums[j - 1] ) ++j;
                    while ( k > j && nums[k] == nums[k + 1] ) --k;
                }
            }
        }
        return ret;
    }
};
/*
时间：68ms，击败：99.66%
内存：19.6MB，击败：97.93%
*/