Leetcode 18. 四数之和

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/junbujianwpl/article/details/88094552

给定一个包含 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]
]

思路:

先排序,然后依次选取最小的数作为已经包含的数,降维,递归,直到只剩下2个数和为指定值,然后左右夹逼,

代码如下:

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        qsort(nums.begin(),nums.end());
        return work(nums,0,nums.size(),target,4);
    }
    
    vector<vector<int>> work(vector<int>& nums, int start, int stop, int dst, int count) {
        vector<vector<int>> result;
        if(count == 0 || start>=stop || nums.size()<count) return result;
        
        if(count == 2){
            int i=start,j=stop-1;
            map<int,int> used;
            while(i<j){
                int sum=nums[i]+nums[j];
                if(used.count(nums[i])!=0){
                    ++i;
                    continue;
                }
                if(sum==dst){
                    result.push_back(vector<int>({nums[i],nums[j]}));
                    used[nums[i]]=1;
                    ++i;
                    --j;
                }else if(sum>dst){
                    --j;
                }else{
                    ++i;
                }
            }
            return result;
        }
        
        map<int,int> used;
        for(int i=start;i<stop;++i){
            if(used.count(nums[i])!=0){
                continue;
            }
            used[nums[i]]=1;
            int newDst=dst-nums[i];
            auto vv=work(nums,i+1,stop,newDst,count-1);
            if(!vv.empty() ){
                for(auto v:vv){
                    v.push_back(nums[i]);
                    result.push_back(v);
                }
            }
        }

        return result;
    }
    
    void qsort(vector<int>::iterator ia, vector<int>::iterator ib){
        if(ia+1>=ib) return;
        
        auto start=ia,sentinel=ib-1;
        while(start<sentinel){
            while(start!=sentinel && *start<=*sentinel) ++start;
            swap(*start,*sentinel);

            while(start!=sentinel && *start<=*sentinel) --sentinel;
            swap(*start,*sentinel);
        }
        
        qsort(ia,sentinel);
        qsort(sentinel,ib);
    }
};

 

展开阅读全文

没有更多推荐了,返回首页