题目:Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.(给定一个含有n个数的数组,找出数组中三个元素a,b,c, 使得a+b+c=0, 找出所以互不重复的满足三数之和为0的数组)
Notice that the solution set must not contain duplicate triplets.(注意不能重复)
Example 1:
Input: nums = [-1,0,1,2,-1,-4]
Output: [[-1,-1,2],[-1,0,1]]
Example 2:
Input: nums = []
Output: []
Example 3:
Input: nums = [0]
Output: []
Constraints:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
题解: 这一题乍一看挺简单,但是仔细一想, 里面有好多坑。首先,不能用三重循环,因为复杂度太高(nnn);其次结果中的数组不能重复,如果在生成所有结果之后再去重,又会增加很多复杂度。
如何避免使用三重循环呢?答: 先排序,每一组结果的三个数也是有序的:
第一个数: 第一个数一定不会大于0,因为如果第一个数都大于0了, 后面两个肯定也大于0,三个数的和就不可能是0.
第一个数一定不会是给定数组的最后两个(否则后面两个数没法取啦)。
第一个数如果有重复, 比如给定的数组是[-1,-1,0,1,-2],那么[-1(第一个),0,1]是一组结果, [-1(第二个),0,1] 也是一组结果, 为了避免结果的重复,需要把第二组去掉。所以判断如果后一个等于前一个 ,就直接continue。
第二个数:第二个数一定在第一个数的后面,同样也需要避免重复。
第三个数:第三个数从后面往前寻找,直到于第二个数相遇,不需要考虑重复的问题,因为如果前两个数都不一样,那么第三个数肯定不一样。
寻找第二个数和第三个数的过程相当于双指针法,一个从前往后一个从后往前。
复杂度: 排序(nlogn)+ 第一个数遍历(n)× 双指针遍历(n), 所以总体复杂度是n平方。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
int size = nums.size();
if(size<3) return result;
std::sort(nums.begin(),nums.end());
int i=0,j,k;
while(nums[i]<=0&&i<size-2)
{
if(i>0 && nums[i-1]==nums[i])
{
i++;
continue;
}
j=i+1;
k=size-1;
while(j<k)
{
if(j>i+1 && nums[j-1]==nums[j])
{
j++;
continue;
}
if(nums[i]+nums[j]+nums[k]==0)
{
vector<int> temp = {nums[i],nums[j],nums[k]};
result.push_back(temp);
j++;k--;
}else if(nums[i]+nums[j]+nums[k]>0)
{
k--;
}else{
j++;
}
}
i++;
}
return result;
}
};