题目描述
给你一个整数数组 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
-10^5 <= nums[i] <= 10^5
分析思路
这道题跟两数之和一个道理,可以用map存储a+b的值和出现的次数,然后遍历最后一个数组,判断c是否等于负的a+b。但是此题要求去重,因此,用map哈希表不方便。
题解中给出了一种双指针的方法。首先对数组进行排序,然后for循环遍历i,给定两个指针left = i+1
和right = i-1
,如果三数之和大于0,则right–,小于0则left++。
思路比较简单,主要是去重的时候需要注意很多的细节,看代码随想录的讲解和视频。
这里去重是因为,首先对数组进行了排序。i去重的原因是:已经将 nums[i - 1] 的所有组合加入到结果中,本次双指针搜索只会得到重复组合。
对left去重的原因是:去掉所有已经用过的left++,对right去重的原因是:吴吊所有已经用过的right–。
看下力扣k神的讲解和动画,也很清楚,k神讲解
class Solution {
public:vector<vector<int>> threeSum(vector<int>& nums) {vector<vector<int>> res;// 排序sort(nums.begin(), nums.end());for(int i=0;i<nums.size();i++){if(nums[i]>0) break;if(i>0 && nums[i]==nums[i-1]){continue;}int left = i+1;int right = nums.size()-1;while(left < right){if(nums[i] + nums[left] + nums[right] > 0) right--;else if(nums[i] + nums[left] + nums[right] <0 ) left++;else{res.push_back(vector<int>{nums[i], nums[left], nums[right]});while(left < right && nums[right]==nums[right-1]) right--;while(left < right && nums[left]==nums[left+1]) left++;left++;right--;}}}return res;}
};