给定一个包含n个整型元素的数组S,其中必定存在三个元素,它们的和为0,找出所有的S中所有满足和为0的唯一三元组
例子:
Input: [-1, 0, 1, 2, -1, -4]
Output:
[
[-1, 0, 1],
[-1, -1, 2]
]
这道题也是3Sum类型,需要找出所有满足条件的三元组,3sum可以考虑:数组排序后,从第一个元素开始,确定第一个,然后在排在其后的有序数组上从两头遍历查找满足条件的数据。
解题思路:
- 首先,先确定一个元素a,a依次从排序后的数组取值
for(int i = 0;i < n -1; i++) a = S[i];
- 其次,设置变量target = -a,在a后的有序数组中寻找和为target的两个元素。
- 按照求有序数组两个数和为指定数的方法求解。
实现:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
vector<vector<int>> result;
for(int i = 0; i < nums.size();i ++)
{
if(i == 0 || nums[i - 1] != nums[i] && i > 0)
{
int target = -nums[i];
int m = i + 1;
int n = nums.size() - 1;
while(m < n)
{
if(nums[m] + nums[n] > target)
{
n --;
}
else if(nums[m] + nums[n] < target)
{
m ++;
}
else
{
vector<int> tmp;
tmp.push_back(nums[i]);
tmp.push_back(nums[m]);
tmp.push_back(nums[n]);
result.push_back(tmp);
while(nums[m] == nums[m + 1] && m < n)
m++;
while(nums[n] == nums[n - 1] && m < n)
n--;
m ++;
n --;
}
}
}
}
return result;
}