版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/fuqiuai/article/details/82972918
题目链接:https://leetcode-cn.com/problems/3sum/description/
题目描述
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解决方法
首先从小到大排序,然后遍历数组中每一个元素,用双指针找到另外两个元素,将三元组存入vector中(注意在此过程中要去重)
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
// 首先从小到大排序,然后遍历数组中每一个元素,用双指针找到另外两个元素,将三元组存入vector中(注意在此过程中要去重)
vector<vector<int>> result;
// 1.排序
sort(nums.begin(),nums.end());
// 2.找到三元组
for (int i=0;i<nums.size();i++){
if (nums[i]>0) break; //剪枝操作,如果当前数大于0则break
if (i>0 && nums[i]==nums[i-1]) continue; //去重操作
int left=i+1, right=nums.size()-1;
while(left<right){
int sum=nums[i]+nums[left]+nums[right];
if (sum>0) right--;
else if (sum<0) left++;
else{
result.push_back({nums[i],nums[left],nums[right]});
while (left<right && nums[left]==nums[left+1]) left++; //这两条语句是为了去重
while (left<right && nums[right]==nums[right-1]) right--;
left++;right--;
}
}
}
return result;
}
};