1、哈希表
基本思路同LeetCode-1,先固定住最左边的位置,而后将题目转化为在一个数组中输出两数之和为0的数字。值得注意的是不同于LeetCode-1,本题中的数字是可以重复的,因此需要考虑更多条件。为了方便起见,我们可以先对整个数组进行排序,而后再进行处理。
考虑到最后输出三元组时要求不重复,我们在循环的时候需要进行注意:1、在最外层循环即确定左值时,需要判断nums[i]与nums[i - 1]是否相同,若相同则说明这种情况已经在上一次循环中解决了,再次计算出的结果一定与上一次的结果相同,因而需要跳出循环;2、在内层循环即确定中值时,需要判断nums[j]、nums[j - 1]与nums[j - 2]三值是否相同。这里考虑的是在确定中值和右值出现的特殊情况——即可能中值与右值相同,但若此时有三个值相同,说明中值和右值组合的情况不止一种,因而会产生重复解需要跳出循环;3、为提高速度使用了哈希集合,在确定了左值之后,我们声明一个哈希集合。若在哈希集合中能找到右值则加入最终的结果中,并删除右值以防再次查询到,若找不到则向哈希集合中添加中值。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++) {
if (nums[i] > 0) {
break;
}
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
unordered_set<int> set;
for (int j = i + 1; j < nums.size(); j++) {
if (j > i + 2 && nums[j] == nums[j - 1] && nums[j - 1] == nums[j - 2]) {
continue;
}
int c = 0 - (nums[i] + nums[j]);
if (set.find(c) != set.end()) {
result.push_back({nums[i], nums[j], c});
set.erase(c);
} else {
set.insert(nums[j]);
}
}
}
return result;
}
};
2、三指针
相较于使用哈希表进行查询,使用三指针进行移动速度会更快。不同于哈希表的地方在于,由于我们已经对数组进行排序,在确定了左指针之后其右边的序列是一个递增序列。考虑到三值之和为0,若此时三值之和大于0则说明中值和右值的和偏大,因此需要将右指针左移;同理当三值之和小于0则说明中值和右值的和偏小,因此需要将中指针右移。当三值之和为0则说明为题目需要的解,此时计入最终结果中,并右移中指针和左移右指针以除去相同的值,最终输出结果。
vector<vector<int>> result;
sort(nums.begin(), nums.end());
for (int left = 0; left < nums.size(); left++) {
if (nums[left] > 0) {
return result;
}
if (left > 0 && nums[left] == nums[left - 1]) {
continue;
}
int middle = left + 1;
int right = nums.size() - 1;
while (right > middle) {
if (nums[left] + nums[middle] + nums[right] > 0) right--;
else if (nums[left] + nums[middle] + nums[right] < 0) middle++;
else {
result.push_back(vector<int>{nums[left], nums[middle], nums[right]});
while (right > middle && nums[right] == nums[right - 1]) right--;
while (right > middle && nums[middle] == nums[middle + 1]) middle++;
right--;
middle++;
}
}
}
return result;