题解 | #牛牛的三元组问题#
牛牛的三元组问题
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)。