题目:找到一组数字中,三个数之和为0的所有数。值得注意的就是不能有重复的组合。
思路:
如果是暴利搜索肯定会超时,所以首先对原数组进行排序,然后开始遍历排序后的数组,这里注意不是遍历到最后一个停止,而是到倒数第三个就可以了,中间如果遇到跟前一个数相同的数字就直接跳过。对于遍历到的数,如果大于0则跳到下一个数,因为三个大于0的数相加不可能等于0;否则用0减去这个数得到一个sum,我们只需要在这个数之后找到两个数之和等于sum即可,这样一来问题又转化为了求two sum,这时候我们一次扫描,找到了等于sum的两数后,加上当前遍历到的数字,按顺序存入结果中即可,然后还要注意跳过重复数字。时间复杂度为 O(n2)。实现代码如下:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res = {};
if (nums.size() < 3) return res;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size() - 2; i++)
{
if (nums[i]>0)//如果当前位大于0,排序后后面也大于0,三个数之和不会等于0
break;
if (i > 0 && nums[i] == nums[i - 1]){ //当前位和前一位相同时,省去继续寻找
continue;
}
int sum = -nums[i];
int left = i + 1, right = nums.size() - 1;
while (left < right)
{
if (nums[left] + nums[right] < sum)
{
while ((left < right) && (nums[left++] == nums[left]))//跳过重复的数字
{
}
}
else if (nums[left] + nums[right] > sum)
{
while ((left < right) && (nums[right--] == nums[right]))//跳过重复的数字
{}
}
else
{
vector<int> three;
three.push_back(nums[i]);
three.push_back(nums[left]);
three.push_back(nums[right]);
res.push_back(three);
while ((nums[left ++] == nums[left])&&(nums[right--]==nums[right]))
{
}
}
}
}
return res;
}
};