题目大意
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
解题思路
解题思路同两数之和。首先对数组排序并固定第一个数字,然后如同两数之和。
注意去重。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int>>ans;
for (int i = 0; i < nums.size(); i++)
{
if (i == 0 || nums[i] != nums[i - 1])
{
int left = i + 1, right = nums.size() - 1, target = 0 - nums[i];
while (left < right)
{
if (nums[left] + nums[right] > target)
--right;
else if (nums[left] + nums[right] < target)
++left;
else
{
if (left == i + 1 || nums[left] != nums[left - 1])
ans.push_back({ nums[i], nums[left], nums[right] });
++left;
--right;
}
}
}
}
return ans;
}
};