给定一个包含 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) { vector<vector<int>> res; sort(nums.begin(),nums.end()); //对于数组进行升序排序 if(!nums.empty() && nums.back() < 0) return {}; //如果数组nums不为空且数组的最后一项小于0,直接返回空 for(int i = 0; i < nums.size(); i++){ if(nums[i] > 0) break; //如果数组开始第一个数字大于0,那么后面的数字都是大于0,直接循环停止 if(i > 0 && nums[i] == nums[i-1]) continue; //如果数组前一个和后一个相等的话,直接跳过后一个,防止重复 int target = 0 - nums[i]; int m = i + 1; //左指针为m int n = nums.size() - 1; //右指针为n while(m < n){ if(nums[m] + nums[n] == target){ res.push_back({nums[i],nums[m],nums[n]}); //如果nums[m]+nums[n]=target,直接记录三个数字的值 while(m < n && nums[m] == nums[m+1]) m++; //如果m<n且后一个和前一个数字相等,那么m++ while(m < n && nums[n] == nums[n-1]) n--; //如果m<n且后一个和前一个数字相等,那么n-- m++; //如果以上条件不满足,直接将m++,n-- n--; } else if(nums[m] + nums[n] < target) m++; //如果两数相加之和小于target,则将左指针m++ else n--; //如果两数相加之和大于target,则将右指针n-- } } return res; }};