<18>——4Sum

来源:互联网 发布:外国人来中国知乎 编辑:程序博客网 时间:2024/05/23 22:18

18、4Sum

四数之和

给一个包含n个数的整数数组S,在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。

 注意事项

四元组(a, b, c, d)中,需要满足a <= b <= c <= d

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

样例

例如,对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为:

(-1, 0, 0, 1)

(-2, -1, 1, 2)

(-2, 0, 0, 2)

分析:在三元组的情况下增加了一组循环。

代码:

class Solution {public:    vector<vector<int>> fourSum(vector<int>& nums, int target) {        sort(nums.begin(),nums.end());        vector<vector<int>> res;        if(nums.size()<4)return res;        for(int i=0;i<nums.size()-3;i++)        {            for(int j=i+1;j<nums.size()-2;j++)            {                int left=j+1,right=nums.size()-1;                while(left<right)                {                    int sum=nums[i]+nums[j]+nums[left]+nums[right];                    if(sum<target)left++;                    else if(sum>target)right--;                    else                    {                        vector<int> four;                        four.push_back(nums[i]);                        four.push_back(nums[j]);                        four.push_back(nums[left]);                        four.push_back(nums[right]);                        res.push_back(four);                        while(left<right&&nums[left]==four[2])left++;                        while(left<right&&nums[right]==four[3])right--;                    }                }                while(j+1<nums.size()-2&&nums[j+1]==nums[j])j++;            }            while(i+1<nums.size()-3&&nums[i+1]==nums[i])i++;        }        return res;    }};



原创粉丝点击