15. 三数之和
给你一个包含 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,则表示右指针的值过大,需要减小,如果当前的三个数值小于0,证明当前的左指针过小,需要右移,直至等于0.
去重:当前的i的情况下,如果出现相同的left左指针,则跳过当前的左值,去找下一个左值。
当前i遍历后,当下一个i与当前i相同时,跳过下一个i
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
if(nums.size()<3)return {};
int count = nums.size();
sort(nums.begin(),nums.end());
vector<vector<int>> ans;
for(int i = 0;i<count-2;i++){
int left = i+1,right =count-1;
while (left < right) {
int ret = nums[i]+nums[left]+nums[right];
if(ret>0){
right--;
}
else if(ret<0){
left++;
}else{
ans.push_back({nums[i], nums[left], nums[right]});
while(left<right&&nums[left+1]==nums[left]){
left++;
}
left++;
}
}
while(i<count-2&&nums[i+1]==nums[i]){
i++;
}
}
return ans;
}
};