这里写目录标题
- 题目
- 思路
- 代码
题目
思路
题目要求三元组不能重复,如果使用哈希表来做,去重很复杂,而且需要额外的空间,我们这里使用双指针法直接针对数组操作。注意题目要求返回的是二维数组!
最重要的是,我们要对数组进行排序,排序后若数组中,第一个元素大于零,则后续不可能存在和等于零的三元组。
其次是对abc去重:
对a去重:当i大于0并且nums[i] == nums[i - 1]时,说明a重复了,直接结束本次循环。i等于0时,是第一次,所以不用去重。为什么i要大于0因为要保证i - 1大于等于0。为什么不是nums[i] == nums[i -+1]?因为nums[i -+1]是left。
对b和c去重:在保证循环体条件right > left情况下,类似于a,判断b的后一个和c的前一个是否有重复的,如果有,b++或者c–。
代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
//返回二维数组 要求a+b+c=0
vector<vector<int>> res;
//排序
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++) {
//如果排序后数组的第一个值大于零,直接返回
if (nums[i] > 0) {
return res;
}
//对a去重
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
//双指针法,寻找b和c
int left = i + 1;
int right = nums.size() - 1;
while (right > left) {
if (nums[i] + nums[left] + nums[right] > 0) {
right--;
} else if (nums[i] + nums[left] + nums[right] < 0) {
left++;
} else {
res.push_back(vector<int>{nums[i], nums[left], nums[right]});
//对b和c去重
while (right > left && nums[right] == nums[right - 1]) right--;
while (right > left && nums[left] == nums[left + 1]) left++;
//双指针收缩,继续循环
right--;
left++;
}
}
}
return res;
}
};