leetcode题库——四数之和

版权声明: https://blog.csdn.net/Dorothy_Xue/article/details/83242227

题目描述:

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

方法:

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        vector<int> flag;
        if(nums.size()==0||nums.size()==1||nums.size()==2||nums.size()==3) return ans;
        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 flagl=j+1,flagr=nums.size()-1;
                while(flagl<flagr)
                {
                    if(flagl>j+1&&nums[flagl]==nums[flagl-1])
                    {
                        flagl++;
                        if(flagl>=flagr) 
                            break;
                        continue;
                    }
                    if(flagr<nums.size()-1&&nums[flagr]==nums[flagr+1])
                    {
                        flagr--;
                        if(flagl>=flagr) 
                            break;
                        continue;
                    }
                    if(nums[i]+nums[j]+nums[flagl]+nums[flagr]==target)
                    {
                        flag.push_back(nums[i]);
                        flag.push_back(nums[j]);
                        flag.push_back(nums[flagl]);
                        flag.push_back(nums[flagr]);
                        ans.push_back(flag);
                        flag.clear();
                        flagl++;
                        continue;
                    }
                    if(nums[i]+nums[j]+nums[flagl]+nums[flagr]>target)
                        flagr--;
                    if(nums[i]+nums[j]+nums[flagl]+nums[flagr]<target)
                        flagl++;
                }
            }
        }
        return ans;
    }
};

 思路:

与三数之和一样。

猜你喜欢

转载自blog.csdn.net/Dorothy_Xue/article/details/83242227