面试题
给定一个包含n个整数的数组nums,判断nums中是否存在三个元素a,b,c ,使得 a + b + c = 0?找出所有满足条件且不重复的三元组。
示例
例如, 给定数组nums = [-1, 0, 1, 2, -1, -4]
满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
题解
#include<iostream>
#include<vld.h>
#include<vector>
#include<algorithm>
using namespace std;
class solution
{
public:
vector<vector<int>> threesums(vector<int>& nums)
{
vector<vector<int>> result;
int length = nums.size();
if (length < 3)//元素小于三个直接返回
return result;
sort(nums.begin(), nums.end());//按升序排列nums数组
for (int i = 0; i < length - 2; i++)//固定位i要留出begin和end的位置,总次数length-2
{
int left = i + 1;
int right = length - 1;
if (nums[i] > 0)
{
break;
}
if (i > 0 && nums[i] == nums[i - 1])//从第二次循环开始,如果固定位nums[i]和上一次相同,则跳过此次循环,目的是去重
{
continue;
}
while (left < right)
{
int sum = nums[i] + nums[left] + nums[right];
if (sum == 0)
{
result.push_back({ nums[i],nums[left],nums[right] });
while (left < right && nums[left] == nums[left + 1])//去重
left++;
while (left < right && nums[right] == nums[right - 1])//去重
right--;
}
else if (sum < 0)
left++;
else
right--;
}
}
return result;
}
};