题目链接:https://leetcode-cn.com/problems/3sum/
解题步骤:
从题意上看,暴力的解法就是使用三重循环遍历数组,判断a + b + c是否等于0,如果等于则将这个组合放进三元组集合中。但是使用这种解法提交时,会超出时间限制,而且很难满足题目中提到的答案中不可以包含重复的三元组的条件。
基本思路都是枚举数组中的三个数,判断相加是否等于0,可以选确定a值,再枚举b、c:
- 对数组进行排序
- 选取第i个元素nums[ i ],0 <= i <= nums.size() - 2 && nums[i] <= 0
- 初始化双指针 left = i + 1, right = nums.size() - 1
- 判断nums[ i ] + nums[ left ] + nums[ right] == 0 ?
- 移动left或rigth
C++实现:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int size = nums.size();
vector<vector<int>> res;
int left, right;
sort(nums.begin(), nums.end()); // 数组升序
for (int i = 0; i < size - 2; i++) {
if (nums[i] > 0) // 如果nums[i] > 0, 则后面的数都 > 0,不需要再遍历
return res;
if (i > 0 && nums[i] == nums[i-1]) // 过滤重复项
continue;
left = i + 1; // 双指针初始化
right = size - 1;
while(left < right) {
if (nums[i] + nums[left] + nums[right] == 0) { // 满足条件
res.push_back({nums[i], nums[left], nums[right]});
while(left < right && nums[right] == nums[right-1]) // 过滤重复项
right -= 1;
while(left < right && nums[left + 1] == nums[left]) // 过滤重复项
left += 1;
right -= 1; // 更新双指针
left += 1;
}
else if (nums[i] + nums[left] + nums[right] < 0) { // < 0 则将lef右移,整体值增加
left += 1;
}
else { // > 0 则将right左移,整体值减少
right -= 1;
}
}
}
return res;
}
};