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;
}