LeetCode 15 3Sum 暴力 C++实现

最新推荐文章于 2024-02-22 19:34:08 发布

原创最新推荐文章于 2024-02-22 19:34:08 发布 · 319 阅读

0 ·

CC 4.0 BY-SA版权

暴力专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一种解决三数之和问题的有效算法。通过先排序再使用双指针技巧，实现时间复杂度为O(n^2)的解决方案。避免了传统三重循环带来的高时间消耗。

传送门：题目

题解：

暴力做，先把序列sort一遍，然后，设三个指针(l，m，r)(左中右)，l从[1:len-2)遍历，然后m从i+1处开始遍历，同时r从len-1处开始遍历。现在的时间复杂度O( $log_2 n+n^2$ )
如果不设三指针，就三个for循环，扫一遍就好，这样的时间复杂度O( $n^3$ )

AC代码：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        int l, m, r, len, sum;
        len = nums.size();
        sort(nums.begin(), nums.end());
        for (l = 0; l < len - 2 && nums[l] <= 0; l++) {
            if (l > 0 && nums[l] == nums[l - 1]){
                continue;
            }
            m = l + 1, r = len - 1;
            while (m < r) {
                sum = nums[l] + nums[m]+ nums[r];
                if (sum == 0)
                    ans.push_back({nums[l], nums[m], nums[r]});
                if (sum <= 0) {
                    m++;
                    while (nums[m - 1] == nums[m] && m < r)
                        m++;
                }
                if (sum >= 0) {
                    r--;
                    while (nums[r + 1] == nums[r] && m < r)
                        r--;
                }
            }
        }
        return ans;
    }
};