原题链接:https://leetcode-cn.com/problems/3sum/
排序+双指针法
- 先将数列从小到大排序
- 先选择一个数,这一步时间复杂度为O(n)
- 在这个数后面的数中用双指针分别从头尾遍历,找到符合条件的数,直到头尾指针相遇,这一步时间复杂度为O(n)
- 循环执行第2、3步,总时间复杂度为O(n*n)
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int n = nums.size();
vector<vector<int>> res;
sort(nums.begin(), nums.end());
for (int i = 0; i < n; i++ ) {
if (i != 0 && nums[i] == nums[i - 1]) continue;
int target = -nums[i];
int j = i + 1, k = n - 1;
while (j < k) {
if (j != i + 1 && nums[j] == nums[j - 1]) {
j++ ;
continue;
}
if (k != n - 1 && nums[k] == nums[k + 1]) {
k-- ;
continue;
}
int sum = nums[j] + nums[k];
if (sum == target) {
res.push_back({nums[i], nums[j], nums[k]});
j++ ;
k-- ;
}
else if (sum < target) j++;
else if (sum > target) k--;
}
}
return res;
}
};