题目描述
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0
且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]] 解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。 nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。 nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。 不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1] 输出:[] 解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0] 输出:[[0,0,0]] 解释:唯一可能的三元组和为 0 。
提示:
3 <= nums.length <= 3000
-100000 <= nums[i] <= 100000
双指针法
思路:
如图所示:先将数组进行升序排列,在一个for循环内,定义两个指针一个左指针为i+1;一个右指针为nums.size()-1;在循环遍历数组时,找到和为0的三元组。
由于是升序排列 ,所以会有以下现象:
- 当nums[i] > 0时,就不会出现nums[i] + nums[left] + nums[right] = 0的情况;
- 当nums[i] + nums[left] + nums[right] > 0 时就说明我们需要减小和的值 ----- 只需要right--即可;
- 当nums[i] + nums[left] + nums[right] < 0 时就说明我们需要增加和的值 ----- 只需要left++即可;
NOTE:因为题目要求答案中不能出现重复的三元组,所以需要对 i left right 分别去重(防止出现重复)
对于i的去重
if(i > 0 && nums[i] == nums[i-1]) {
continue;
}
if( nums[i] == nums[i+1]) {
continue;
}
这两个是不一样的
对于这个[-1,-1,2] 数组 ,如果使用 nums[i] == nums[i+1] 去重 那么就会直接忽略掉 [-1,-1,2]这个正确的三元组;当i移动到第二个-1时,就会判断 nums[i] == nums[i+1] 直接跳过
而对于i > 0 && nums[i] == nums[i-1] 这个条件来说 就相当先对于i=0去查找符合的答案 再进行去重,避免出现遗漏答案的情况,
查看上面这个数组[-1,-1,3,3]
如果有答案出现 在第一个-1的时已经获取 ,没有答案出现时 第一个-1也已经循环遍历过了,到了第二个-1也是一样的结果,直接跳过
这与下面的left , right去重的原理类似,在得到一个正确答案的前提下去重,要不然去除的是答案中的相同元素;
排序
在C++中,sort()
函数用于对数组或容器中的元素进行排序。sort()
函数是标准模板库(STL)中的一部分,定义在 <algorithm>
头文件中
基本用法
sort()
函数可以对给定范围内的所有元素进行排序,默认情况下,它按照升序排列元素。其基本形式如下:
sort(RandomAccessIterator first, RandomAccessIterator last);
- 参数:
first
: 范围的起始迭代器。last
: 范围的结束迭代器(指向最后一个元素之后的位置,符合C++的半开区间规则)。
自定义比较函数
你可以提供自己的比较函数来定义不同的排序规则,比如降序排列或按特定标准排序。比较函数需要定义为第三个参数:
sort(RandomAccessIterator first, RandomAccessIterator last, Compare comp);
- 参数:
comp
: 一个可调用的比较函数或者函数对象,它接受两个元素作为参数,返回一个bool
值。当第一个参数应该排在第二个参数之前时,应返回true
。
#include <iostream>
#include <vector>
#include <algorithm>
bool compare(int a, int b) {
return a > b; // 降序排列
}
int main() {
std::vector<int> nums = {4, 1, 3, 9, 7};
sort(nums.begin(), nums.end(), compare); // 使用自定义的比较函数
for (int num : nums) {
std::cout << num << " ";
}
std::cout << std::endl;
return 0;
}
完整代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
//sort()默认升序排列
sort(nums.begin() , nums.end());
for (int i = 0; i < nums.size() ;i++) {
//如果出现nums[i] > 0 直接返回,不会出现答案了
if(nums[i] > 0) {
return result;
}
// 错误去重a方法,将会漏掉-1,-1,2 这种情况
/*
if (nums[i] == nums[i + 1]) {
continue;
}
*/
//对i去重 先对i=0 进行查找答案
if(i > 0 && nums[i] == nums[i-1]) {
continue;
}
int left = i+1;
int right = nums.size() -1;
while(left < right) {
// 去重复逻辑如果放在这里,0,0,0 的情况,可能直接导致 right<=left 了,从而漏掉了 0,0,0 这种三元组
/*
while (right > left && nums[right] == nums[right - 1]) right--;
while (right > left && nums[left] == nums[left + 1]) left++;
*/
if(nums[i] + nums[left] + nums[right] > 0) {
right--;
}
else if(nums[i] + nums[left] + nums[right] < 0) {
left++;
}
else {
//获得答案 存入result二维数组中
result.push_back(vector<int>{nums[i],nums[left],nums[right]});
//对left和right去重
while(right > left && nums[left] == nums[left+1]) {
left++;
}
while(right > left && nums[right] == nums[right-1]) {
right--;
}
//得到答案要更新right和left
right--;
left++;
}
}
}
return result;
}
};