leetcode——18.四数之和

最新推荐文章于 2025-07-27 17:01:48 发布

原创最新推荐文章于 2025-07-27 17:01:48 发布 · 91 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

145 篇文章

订阅专栏

本文介绍了一个解决四数之和问题的高效算法。通过先排序再使用三层循环结合双指针技巧，避免了重复解并降低了搜索空间，实现O(n³)的时间复杂度。

思路

在三数之和基础上加多了一层遍历

代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        sort(nums.begin(),nums.end());
        int n=nums.size();
        for(int first=0;first<n-1;first++){
        	//保留重复数组中的第一个元素
            if(first>0&&nums[first]==nums[first-1])
                continue;
            for(int second=first+1;second<n;second++){
            	//保留重复数组中的第一个元素
                if(second>first+1&&nums[second]==nums[second-1])
                    continue;
                //转为两数之和问题 双向指针遍历
                int third=second+1,fourth=n-1;
                while(third<fourth){
                    int sum=nums[first]+nums[second]+nums[third]+nums[fourth];
                    if(sum>target)
                        fourth--;
                    else if(sum<target)
                        third++;
                    else{
                        ans.push_back({nums[first],nums[second],nums[third],nums[fourth]});
                        //移动third fourth指针 跳过重复元素
                        int third_val=nums[third];
                        int fourth_val=nums[fourth];
                        do{third++;}
                        while(third<fourth&&third_val==nums[third]);
                        do{fourth--;}
                        while(third<fourth&&fourth_val==nums[fourth]);
                    }
                }
            }
        }
        return ans;
    }
};