题目:
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:
暴力求解需要三重循环时间复杂度较高,这里可以转化为求解两数之和。首先将数组从小到大进行排序,然后从第一个元素开始遍历所有的元素,每次从被访问的元素之后的数组里面找出两个数使得这三个数的和为0,这就是两数之和思路的拓展。注意,要避免重复的情况出现,当被访问的元素和前面的元素相等时,可以跳过该元素,直接访问下面的元素。还有求后面的两个数的时候也是一样,遇到重复的数字就跳过。还有一个要注意的地方就是当第一个元素大于0的时候,后面的数都会大于0,它们的和就不可能为0了,所以这个时候要终止算法。
代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>>result;
int n = nums.size();
if (n < 3) return result;
sort(nums.begin(), nums.end());
vector<int>temp(3);
for (int i = 0; i < n; i++)
{
if (nums[i] > 0) break;
if (i > 0 && nums[i] == nums[i - 1]) continue;
int begin = i + 1, end = n - 1;
while (begin<end)
{
int sum = nums[i] + nums[begin] + nums[end];
if (sum == 0)
{
temp[0] = nums[i];
temp[1] = nums[begin];
temp[2] = nums[end];
result.push_back(temp);
begin++;
end--;
while (begin < end && nums[begin] == nums[begin - 1]) begin++;
while (begin < end && nums[end] == nums[end + 1]) end--;
}
else if (sum > 0) end--;
else begin++;
}
}
return result;
}
};