给定一个包含 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] ]
此题与15题求三数之和和16题求与三数之和最接近的思路是一样,都要转化为求两数之和的问题,降低复杂度。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> vec;
vector<int> v(4);
if(nums.size()<4) return vec;
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size()-3;++i){
for(int j=i+1;j<nums.size()-2;++j){
int m=target-nums[i]-nums[j];//转化成求两数之和的问题,接下来的解法与求三数之和时一样
int l=j+1,r=nums.size()-1;
while(l<r){
int sum=nums[l]+nums[r];
if(m==sum){
v[0]=nums[i];
v[1]=nums[j];
v[2]=nums[l];
v[3]=nums[r];
vec.push_back(v);
++l;
--r;
while(l<r&&nums[r]==nums[r+1]) --r;//注意去重的位置在此
while(l<r&&nums[l]==nums[l-1]) ++l;
}
else if(m<sum){
--r;
}
else{
++l;
}
}
while(j<nums.size()-2&&nums[j]==nums[j+1]) ++j;
}
while(i<nums.size()-3&&nums[i]==nums[i+1]) ++i;
}
return vec;
}
};