给定一个包含 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>> ret;
int n=nums.size();
if(n<4)
return ret;
sort(nums.begin(),nums.end());
for(int i=0;i<=n-4;i++)
{
for(int j=i+1;j<=n-3;j++)
{
int m=j+1;
int k=n-1;
while(m<k)
{
if(nums[i]+nums[j]+nums[m]+nums[k]<target)
m++;
else if(nums[i]+nums[j]+nums[m]+nums[k]>target)
k--;
else
{
ret.push_back({nums[i],nums[j],nums[m],nums[k]});
m++;
k--;
while(j<k&&nums[m]==nums[m-1])
m++;
while(j<k&&nums[k]==nums[k+1])
k--;
}
}
while(nums[j]==nums[j+1])
j++;
}
while(nums[i]==nums[i+1])
i++;
}
return ret;
}
};