解题思路:将数组排序,枚举一个数,剩下两个数在数组中用 Leetcode 001 中的方法找到,移动指针的时候需要注意相同数字的情况。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ret;
sort(nums.begin(), nums.end());
int n = nums.size(), i = 0;
while (i < n) {
int a = nums[i];
int j = i + 1, k = n - 1;;
while (j < k) {
int b = nums[j], c = nums[k];
int s = a + b + c;
if (s == 0) ret.push_back({a, b, c});
if (s <= 0) while (j < k && nums[j] == b) j++;
if (s >= 0) while (j < k && nums[k] == c) k--;
}
while (nums[i] == a && i < n) i++;
}
return ret;
}
};