题解 | #牛牛的三元组问题#

牛牛的三元组问题

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;
    }
};

这是一个典型的三数之和问题,可以使用双指针法来解决。

具体思路如下:

  1. 首先对数组进行排序,方便后续操作。
  2. 遍历排序后的数组,将当前元素作为第一个数,然后使用双指针法在剩余的数组中寻找另外两个数,使得三个数的和为0。
  3. 双指针法的具体操作如下:定义两个指针,一个指向当前元素的下一个位置,一个指向数组末尾。如果三个数的和等于0,则将这三个数加入结果数组中,并将左指针右移,右指针左移,继续寻找下一个满足条件的三元组。如果三个数的和小于0,则将左指针右移。如果三个数的和大于0,则将右指针左移。
  4. 遍历结束后,返回结果数组。

时间复杂度分析: 排序的时间复杂度为O(nlogn),外层循环的时间复杂度为O(n),内层循环的时间复杂度为O(n),因此总的时间复杂度为O(n^2)。

空间复杂度分析: 除了返回结果的二维数组外,只需要常数级别的额外空间,因此空间复杂度为O(1)。

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务