这里写目录标题
- 题目
- 思路
- 代码
题目
思路
题目要求三元组不能重复,如果使用哈希表来做,去重很复杂,而且需要额外的空间,我们这里使用双指针法直接针对数组操作。注意题目要求返回的是二维数组!
最重要的是,我们要对数组进行排序,排序后若数组中,第一个元素大于零,则后续不可能存在和等于零的三元组。
其次是对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=0vector<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和cint 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;}
};