leetcode 18. 四数之和

给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c +d 的值与 target 相等?找出所有满足条件且不重复的四元组。

注意:

答案中不可以包含重复的四元组。

示例:

给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

满足要求的四元组集合为:
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

此题与15题求三数之和和16题求与三数之和最接近的思路是一样,都要转化为求两数之和的问题,降低复杂度。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> vec;
        vector<int> v(4);
        if(nums.size()<4) return vec;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size()-3;++i){
            for(int j=i+1;j<nums.size()-2;++j){
                int m=target-nums[i]-nums[j];//转化成求两数之和的问题,接下来的解法与求三数之和时一样
                int l=j+1,r=nums.size()-1;
                while(l<r){
                    int sum=nums[l]+nums[r];
                    if(m==sum){
                        v[0]=nums[i];
                        v[1]=nums[j];
                        v[2]=nums[l];
                        v[3]=nums[r];
                        vec.push_back(v);
                        ++l;
                        --r;
                        while(l<r&&nums[r]==nums[r+1]) --r;//注意去重的位置在此
                        while(l<r&&nums[l]==nums[l-1]) ++l;
                    }
                    else if(m<sum){
                        --r;
                    }
                    else{
                        ++l;
                    }
                }
                while(j<nums.size()-2&&nums[j]==nums[j+1]) ++j;
            }
            while(i<nums.size()-3&&nums[i]==nums[i+1]) ++i;
        }
        return vec;
    }
};

猜你喜欢

转载自blog.csdn.net/u011750466/article/details/80260308