(写给未来遗忘的自己)
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int>> ternary_number;
if (nums.size() < 3) {
return ternary_number;
}
for (int i = 0; i < nums.size() - 2; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int first_ptr = i + 1;
int last_ptr = nums.size() - 1;
while (first_ptr<last_ptr) {
int sum = nums[i] + nums[first_ptr] + nums[last_ptr];
if (sum > 0) {
last_ptr--;
} else if (sum < 0) {
first_ptr++;
} else {
ternary_number.push_back(vector<int>{nums[i], nums[first_ptr], nums[last_ptr]});
while (first_ptr < last_ptr && nums[first_ptr] == nums[first_ptr + 1]) {
first_ptr++;
}
while (first_ptr < last_ptr && nums[last_ptr] == nums[last_ptr - 1]) {
last_ptr--;
}
first_ptr++;
last_ptr--;
}
}
}
return ternary_number;
}
};
这个和四数之和等于零的区别在于,同一个数组里面,三个数的下标要求完全不一样。
思路:
一. 3个数的下标保证不相同
1.将数组排好序,之后固定遍历指针取第一个,求解三数之和
2.三数之和大于0,则右侧指针左移,如果小于0,则左侧指针右移。
3.直到左右指针碰面结束循环(碰面继续走下去出现的结果就是first_ptr和last_ptr互换的结果)。4.然后遍历指针下移一位,重复上述步骤。
二. 3个元组数和以前的不一样(这一个写在注意里面,可能容易忽略)
(我们始终记住数组是排好序的,就是从小到大的)
我们需要对三个数去重:
第一个数:如果和上一个数一样,那么后面的两个数只要和上一次一样就可以,就会出现三个数重复。(0,3,4)(1,3,4)就会出现下标一样的数组。由于我们上面说的是排好序的,所以只要挨着的不一样,之后一定不一样。
第二个和第三个数:当出现一组合适的三元数后,第一个数目前还是保持不变的(-2),如果第二个和第三个数出现连续一样的数,那么加进去就是一样的三元数。所以去重。(1,3,5)(1,4,6)虽然下标不一样,但是出来的三元数一样,所以需要去掉。