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 ≤ n u m s . l e n g t h ≤ 3000 0 \le nums.length \le 3000 0≤nums.length≤3000
- − 1 0 5 ≤ n u m s [ i ] ≤ 1 0 5 -10^5 \le nums[i] \le 10^5 −105≤nums[i]≤105
题解:
排序 + 双指针。
首先对 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(n^2) O(n2)
额外空间复杂度: O ( 1 ) O(1) 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%
*/