力扣:15.三数之和
题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
解析:
首先排序,然后将三个值分为两个值,a、b+c
用三指针,一个指针作为初始值a,left 指针作为b,right 指针作为c。
a指向第一个位置,left指向第二个位置,right指向末尾,然后将 left+right的值与 b+c 的标准值比较,left+right大则向左移动right,小则向右移动 left。若存在则放入容器中继续移动直到right = left。然后a指向第二个位置(小于0)继续下去。
if (first > 0 && nums[first] == nums[first - 1])
first-1 要存在,即至少要从起始的第二个值开始,所以 first 要大于0。
去重问题:
重复的原因:例:-101满足条件。出现重复问题时要么-1重复,要么0重复,要么1重复。只需要查找到满足条件的值之后,此时若选定了 first 且未结束,分别保证second、third不会重复出现即可。若first要进入下一轮保证first不重复即可。first判断重复用 nums[first] == nums[first - 1],判断当前值跟前一个值(已判断可行性)是否相同,不是判断当前值跟后一个值是否相同。
代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>>v;
sort(nums.begin(), nums.end());
for (int first = 0; first < nums.size(); first++){
if (nums[first] >0) return v;
if (first > 0 && nums[first] == nums[first - 1]) continue;
int second = first + 1;
int third = nums.size() - 1;
while (third > second) {
if (nums[first]+nums[second]+nums[third]> 0) third--;
else if (nums[first]+nums[second]+nums[third] < 0) second++;
else {
v.push_back(vector<int>{nums[first],nums[second],nums[third]});
while(second < third && nums[second] == nums[second + 1]) second++;
while(second < third && nums[third] == nums[third - 1]) third--;
second++;
third--;
}
}
}
return v;
}
};