力扣(LeetCode)18. 四数之和(C++)

最新推荐文章于 2022-12-23 13:30:18 发布

清墨韵染

最新推荐文章于 2022-12-23 13:30:18 发布

阅读量506

点赞数 1

分类专栏：墨染leetcode 文章标签： c++ leetcode 算法模拟双指针

本文链接：https://blog.csdn.net/Innocence02/article/details/127940504

版权

墨染leetcode 专栏收录该内容

138 篇文章 2 订阅

订阅专栏

双指针

快排使 $n u m s$ 正序。

设置四个指针 $i$ 指向 $n u m s$ 第一个数， $j$ 指向 $n u m s$ 第二个数，从前往后枚举 $n u m s [i]$ 和 $n u m s [j]$ ， $l$ 从 $n u m s [j + 1]$ 往后，指向第三个数， $r$ 从 $n u m s . s i z e () - 1$ 往左，指向第四个数。

设四数之和 $n u m s [i] + n u m s [j] + n u m s [l] + n u m s [r] = s u m$ ，如果 $s u m > t a r g e t$ ，为了接近 $t a r g e t$ ， $r - -$ 。同理， $s u m < t a r g e t$ ，为了接近 $t a r g e t$ ， $l + +$ 。

如果 $s u m = t a r g e t$ 得到一个答案，为了避免重复答案， $l$ 循环右移，直到遇到不重复元素，同理 $r$ 循环左移。

代码展示

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        int n = nums.size();
        vector<vector<int>> ans;
        if(nums.size()<4) return ans;
        sort(nums.begin(),nums.end());
        for(int i = 0;i<n-3;i++){
            if(i&&nums[i]==nums[i-1]) continue;
            for(int j = i+1;j<n-2;j++){
                if(j!=i+1&&nums[j]==nums[j-1]) continue;
                int l = j+1,r = n - 1;
                while(l<r){
                    long long sum = (long long)nums[i]+nums[j]+nums[l]+nums[r];
                    if(sum>target) r--;
                    else if(sum<target) l++;
                    else{
                        ans.push_back({nums[i],nums[j],nums[l],nums[r]});
                        while(l<r&&nums[l]==nums[l+1]) l++;
                        while(l<r&&nums[r]==nums[r-1]) r--;
                        l++;
                        r--;
                    }
                }
            }
        }
        return ans;
    }
};

博主致语

理解思路很重要！
欢迎读者在评论区留言，作为日更博主，看到就会回复的。

AC

复杂度分析

时间复杂度: $O(n^3)$ ， $n$ 是 $n u m s$ 的长度，枚举 $n u m s [i]$ 和 $n u m s [j]$ ，同时双指针枚举 $n u m s [l] / n u m s [r]$ 的时间复杂度 $O(n^3)$ 。
空间复杂度: $O (l o g n)$ ，快排的平均空间复杂度 $O (l o g n)$ 。