链接:戳这里
题意:给出长度为n的整数序列,要求找出满足条件的所有不同三元组使得三个数和为0
两个三元组不相同意义为:集合中至少有一个数不同
思路:序列按升序排序,枚举a[k],接着在[0,k-1]选出两个数a[i],a[j] 相加使得和为-a[k],即为答案 复杂度O(n*n)
由于序列是升序排序,所以a[i]+a[j]的值是会随着i,j的移动而显得有单调性
当a[i]+a[j]<-a[k]:则i++ (当前的a[i]==a[i-1]的话,说明不需要再去比较i了 直接i++ 去重
当a[i]+a[j]>-a[k]:则j-- (当前的a[j]==a[j+1]的话,说明不需要再去比较j了,直接j-- 去重
相等的话统计答案,接着i++,j--
代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int n=nums.size();
sort(nums.begin(),nums.end());
vector<vector<int> > ans;
for(int k=2;k<n;k++){
if(k+1<n && nums[k]==nums[k+1]) continue;
int i=0,j=k-1;
int tmp=-nums[k];
while(i<j){
int sum=nums[i]+nums[j];
if(sum<tmp || (nums[i]==nums[i-1] && i>0) ) i++;
else if(sum>tmp || (nums[j]==nums[j+1] && j+1<k) ) j--;
else ans.push_back({nums[i++],nums[j--],nums[k]});
}
}
return ans;
}
};