给定一个包含 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;
sort(nums.begin(),nums.end());
if(nums.size()<4) // 若数组长度小于4,返回空ret
return ret;
for(int i=0;i<nums.size();i++){
if(i&&nums[i]==nums[i-1]) // 第一个数的去重
continue;
for(int j=i+1;j<nums.size();j++){
if(j>i+1&&nums[j]==nums[j-1]) // 第二个数的去重
continue;
int t1=j+1,t2=nums.size()-1; // 双动态指针
while(t1<t2){
vector<int> v; // 注意v的位置,易错!!!
int sum=nums[i]+nums[j]+nums[t1]+nums[t2]; // 四数之和
if(sum==target){
v.push_back(nums[i]);
v.push_back(nums[j]);
v.push_back(nums[t1]);
v.push_back(nums[t2]);
while(t1<t2&&nums[t1]==nums[t1+1]) t1++; // 第三个数的去重,不需使用unique函数
while(t1<t2&&nums[t2]==nums[t2-1]) t2--; // 第四个数的去重
t1++;
t2--;
ret.push_back(v);
}
else if(sum<target)
t1++;
else
t2--;
}
}
}
return ret;
}
};