题目描述
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
样例
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
if(nums.size() < 3) return res; //判断数组不足3个的情况
sort(nums.begin(),nums.end()); //排序下 nlogn
/*
考虑重复情况 不妨让i < j < k,你拥有一个排序好的数组,
选定第一个i, 然后需要另外两个数加起来等于 -i
为了保持升序 j 从 i+1 开始遍历,然后只需要找到k了,
*/
for(int i = 0; i < nums.size(); i ++) {
//为了保证不重复 i,j的值都需要和上一轮不一样!!!
if(i && nums[i] == nums[i-1]) continue;
for(int j = i + 1, k = nums.size()-1; j < k; j ++) {
if(j != i + 1 && nums[j] == nums[j-1]) continue;
while(j < k - 1 && nums[i]+nums[j]+nums[k] > 0) k--;
//你可以从将k从右向左遍历 直到找到第一个使得三数之和小于等于0的k
//这时判断下 三数之和是否等于0
if(nums[i]+nums[j]+nums[k] == 0) res.push_back({nums[i],nums[j],nums[k]});
//最关键的是 下一次循环的时候k回退, 因为j的数值一定会变大,为了找到0值,k一定是保持不变或变小
}
}
return res;
}
};