15三数之和
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
分析:这题注意解决超时和去去重两个问题
解答:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
if(nums.size()<3) return vector<vector<int>>();
vector<vector<int>> res;
vector<int> threeSum(3,0);
sort(nums.begin(),nums.end());
for(int left=0;left<nums.size()-2;left++){
int mid = left+1;
int right = nums.size()-1;
while(mid<right){
int sum = nums[left]+nums[mid]+nums[right];
if(sum<0){
mid++;
}
else if(sum>0){
right--;
}
else{
threeSum[0]=nums[left];
threeSum[1]=nums[mid];
threeSum[2]=nums[right];
res.push_back(threeSum);
while(nums[mid]==nums[mid+1]&&mid+1<right) mid = mid+1;//去重
while(nums[right]==nums[right-1]&&right-1>mid) right=right-1;//去重
mid++;
right--;
}
}
while(nums[left]==nums[left+1]&&left+1<nums.size()-2) left++;//去重
}
return res;
}
};