原题链接:https://oj.leetcode.com/problems/3sum/
经典3sum题,不过要注意这里的去重,去重是给原来这道题的基本形式加的难点。
1. 数组如果大小小于3,那就没有任何结果。
2. 将数组排序
3. 然后循环数组,每次都选取i所在的当前值为第一个结果,然后选取left = i + 1和right = size - 1。则问题变为num[left] + num[right]是否等于-num[i]。
4. 每次选取第一个元素的时候,要将其与之前的第一个元素相比,如果相同,则说明如果继续计算,此轮的计算结果已经被包含在上轮的计算结果之内,所以跳至下一轮。
5. 当left和right满足条件时候,我们需要将left和right移至下一个位置,这里需要将left之后和right之前的重复元素全部跳过,否则将会出现重复答案,则不符合题意。
class Solution {
public:
vector<vector<int> > threeSum(vector<int> &num) {
vector<vector<int> > res;
if (num.size() < 3) {
return res;
}
sort(num.begin(), num.end());
int prev = INT_MIN;
for (int i = 0; i < num.size() - 2; ++i) {
if (num[i] == prev) {
prev = num[i];
continue;
} else {
prev = num[i];
}
int target = -num[i];
int left = i + 1;
int right = num.size() - 1;
while (left < right) {
int curSum = num[left] + num[right];
if (curSum == target) {
vector<int> curRes;
curRes.push_back(num[i]);
curRes.push_back(num[left]);
curRes.push_back(num[right]);
res.push_back(curRes);
while (num[left] == num[left + 1] && left + 1 < num.size()) ++left;
while (num[right] == num[right - 1] && right - 1 >= 0) --right;
++left;
--right;
} else if (curSum < target) {
++left;
} else {
--right;
}
}
}
return res;
}
};