真正见识到了该题的恶心之处,写了1h才改出来,心酸


#include<algorithm>
#include<functional>
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
// i,j,k都不同,相加为0
// 先将数组排序,例如-4 -1 -1 0 1 2
// 固定-4,双指针指向后面滑动依次寻找有无符合者
sort(nums.begin(),nums.end());
vector<vector<int>> three_shuzu;
int length = nums.size();
if(length < 3 || nums[0] > 0) return three_shuzu;
int j=1,k=length-1;
for(int i=0;i<length-2;i++){
if(i > 0 && nums[i] == nums[i-1]) continue;
j=i+1;
k=length-1;
while(j<k){
if(nums[j]+nums[k]+nums[i]==0){
vector<int> p1{nums[i],nums[j],nums[k]};
three_shuzu.push_back(p1);
while(j < k && nums[j] == nums[j+1]) j++;
while(j < k && nums[k] == nums[k-1]) k--;
j++;
k--;
}
//这时候不相等
else if(nums[j]+nums[k]+nums[i]<0){
j++;
// while(j<k&&nums[j+1]==nums[j]){j++;}
}
else{
k--;
// while (j < k && nums[k] == nums[k-1]) k--;
}
}
}
return three_shuzu;
}
};
刚开始想的时候就是排序+双指针滑动,奈何学艺不精,试了好几遍才对。
主要思路就是:
1、将数组排序
2、固定第一位i, j指向i+1,k指向最后
3、sum==0时添加,大于小于时,看代码中解释即可。
后面的else if和else中,我尝试用while掠过相同的,会漏解啊
官方题解:异曲同工之妙吧,反正我是没想到这样的,不过感觉类似,这题思路不难,就是实际写的时候有点恶心。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int n = nums.size();
sort(nums.begin(), nums.end());
vector<vector<int>> ans;
// 枚举 a
for (int first = 0; first < n; ++first) {
// 需要和上一次枚举的数不相同
if (first > 0 && nums[first] == nums[first - 1]) {
continue;
}
// c 对应的指针初始指向数组的最右端
int third = n - 1;
int target = -nums[first];
// 枚举 b
for (int second = first + 1; second < n; ++second) {
// 需要和上一次枚举的数不相同
if (second > first + 1 && nums[second] == nums[second - 1]) {
continue;
}
// 需要保证 b 的指针在 c 的指针的左侧
while (second < third && nums[second] + nums[third] > target) {
--third;
}
// 如果指针重合,随着 b 后续的增加
// 就不会有满足 a+b+c=0 并且 b<c 的 c 了,可以退出循环
if (second == third) {
break;
}
if (nums[second] + nums[third] == target) {
ans.push_back({nums[first], nums[second], nums[third]});
}
}
}
return ans;
}
};
1786

被折叠的 条评论
为什么被折叠?



