给你一个整数数组
nums
,判断是否存在三元组[nums[i], nums[j], nums[k]]
满足i != j
、i != k
且j != k
,同时还满足nums[i] + nums[j] + nums[k] == 0
。请你返回所有和为0
且不重复的三元组。注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]] 解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。 nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。 nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。 不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。示例 2:
输入:nums = [0,1,1] 输出:[] 解释:唯一可能的三元组和不为 0 。示例 3:
输入:nums = [0,0,0] 输出:[[0,0,0]] 解释:唯一可能的三元组和为 0 。提示:
3 <= nums.length <= 3000
-105 <= nums[i] <= 105
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
//获取数组大小
int n =nums.size();
//排序数组,方便后续双指针操作
sort(nums.begin() , nums.end());
vector<vector<int>> ans;//存储结果
//枚举a
for(int first = 0; first < n;++first ) {
//需与上次枚举结果不同,避免重复
if(first >0 && nums[first] == nums[first -1]) {
continue;
}
//c初始对应数组最右端
int third = n-1;
int target = -nums[first];// b+c = -a
//枚举b
for(int second = first +1; second < n ;second++) {
//需与上次枚举结果不同,避免重复
if(second > first +1 && nums[second] == nums[second -1]) {
continue;
}
// 需要保证 b 的指针在 c 的指针的左侧
while(second <third && nums[second] + nums[third] > target) {
--third;//如果当前值大于target,c左移
}
// 如果指针重合,随着 b 后续的增加
// 就不会有满足 a+b+c=0 并且 b<c 的 c 了,可以退出循环
if(second == third) {
break;
}
// 检查是否满足 a + b + c = 0
if(nums[second] + nums[third] == target) {
//将符合的三元组放进ans
ans.push_back({nums[first] , nums[second] , nums[third]});
}
}
}
return ans;
}
};
官方题解 加了注释好理解