题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
本题的难点在于时间复杂度的控制(优化算法)和 (去重)
思路: 控制变量的方式减少运算+排序优化
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums)
{
// 类似控制变量法,同时排序
// 本题的难点是去重
vector<vector<int>> res;
// special input
if (nums.size() <= 2) {
return res;
}
if (nums.size() == 3) {
if (nums[0] + nums[1] + nums[2] == 0) {
res.push_back({nums[0], nums[1], nums[2]});
}
return res;
}
// 多个元素
// 排序
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size()-2; i++) {
// 去重1
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
// 排序后,正数之后不可能出现和为0
if (nums[i] > 0) {
break;
}
// 二分
int l = i + 1;
int r = nums.size() - 1;
while (l < r) {
int sum = nums[i] + nums[l] + nums[r];
if(sum == 0) {
// 去重
while (l < r && nums[l] == nums[l + 1]) {
l++;
}
// 去重
while (l < r && nums[r] == nums[r - 1]) {
r--;
}
res.push_back({nums[i], nums[l], nums[r]});
l++;
r--;
} else if (sum < 0) {
l++;
} else {
r--;
}
}
}
return res;
}
};