Leetcode Problem18

4Sum

问题描述:给定一个数组,要求一个四元组的四个数之和为target,找出所有不重复的四元组。

这道题跟之前的3Sum是同一类型的题,就是在3Sum的基础上再加上一个循环。

vector<vector<int>> fourSum(vector<int>& nums,int target)
{
    vector<vector<int>> res;
    if(nums.size()<4) return res;
    sort(nums.begin(),nums.end());
    for(int i=0;i<nums.size()-3;i++)
    {
        if(i>0&&nums[i]==nums[i-1]) continue; //去除重复
        for(int j=i+1;j<nums.size()-2;j++)
        {
            if(j>i+1&&nums[j]==nums[j-1]) continue; //去除重复
            int m=j+1,n=nums.size()-1;
            while(m<n)
            {
                if(nums[i]+nums[j]+nums[m]+nums[n]==target)
                {
                    res.push_back({nums[i],nums[j],nums[m],nums[n]});
                    m++;n--;
                    while(m<n&&nums[m]==nums[m-1]) m++; //去除重复
                    while(m<n&&nums[n]==nums[n+1]) n--; //去除重复
                }
                else if(nums[i]+nums[j]+nums[m]+nums[n]<target)
                {
                    m++;
                    while(m<n&&nums[m]==nums[m-1]) m++;
                }   
                else
                {
                    n--;    
                    while(m<n&&nums[n]==nums[n+1]) n--;
                }

            }
        }       
    }
    return res;
}

猜你喜欢

转载自blog.csdn.net/vandance/article/details/81878691