题解 | #牛牛的三元组问题#
牛牛的三元组问题
https://www.nowcoder.com/practice/72c6d735fb1144a2ba162976a4510839
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
vector<vector<int> > findTriplets(vector<int>& nums) {
vector<vector<int>> res;
int n = nums.size();
if (n < 3) {
return res;
}
sort(nums.begin(), nums.end());
for (int i = 0; i < n - 2; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1;
int right = n - 1;
while (left < right) {
int sum = nums[i] + nums[left] + nums[right];
if (sum == 0) {
res.push_back({nums[i], nums[left], nums[right]});
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
left++;
right--;
} else if (sum < 0) {
left++;
} else {
right--;
}
}
}
return res;
}
};
这是一个典型的三数之和问题,可以使用双指针法来解决。
具体思路如下:
- 首先对数组进行排序,方便后续操作。
- 遍历排序后的数组,将当前元素作为第一个数,然后使用双指针法在剩余的数组中寻找另外两个数,使得三个数的和为0。
- 双指针法的具体操作如下:定义两个指针,一个指向当前元素的下一个位置,一个指向数组末尾。如果三个数的和等于0,则将这三个数加入结果数组中,并将左指针右移,右指针左移,继续寻找下一个满足条件的三元组。如果三个数的和小于0,则将左指针右移。如果三个数的和大于0,则将右指针左移。
- 遍历结束后,返回结果数组。
时间复杂度分析: 排序的时间复杂度为O(nlogn),外层循环的时间复杂度为O(n),内层循环的时间复杂度为O(n),因此总的时间复杂度为O(n^2)。
空间复杂度分析: 除了返回结果的二维数组外,只需要常数级别的额外空间,因此空间复杂度为O(1)。

查看23道真题和解析