给定一个包含 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]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
完整代码
这道题的难点在于如何去重,并且保证不超时,这里采用集合的方法去重。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
set<vector<int>> res;
vector<vector<int>> result;
if(nums.size()<4)
return result;
vector<int> c;
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!=nums.size()-3&&nums[j]==nums[j-1])
//continue;
int s=j+1;
int e=nums.size()-1;
while(s<e){
int cur=nums[i]+nums[j]+nums[s]+nums[e];
if(cur==target){
c.clear();
c={nums[i],nums[j],nums[s],nums[e]};
res.insert(c);
++s;
// while(s<e&&nums[s]==nums[s-1])
// ++s;
}
else if(cur>target){
--e;
// while(s<e&&nums[e]==nums[e+1])
// --e;
}
else{
++s;
// while(s<e&&nums[s]==nums[s-1])
// ++s;
}
}
}
}
for(auto c:res)
result.push_back(c);
return result;
}
};
如何利用跳过重复数字的方法去重呢???
其实是可以的,参见代码
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> res;
if(nums.size()==0)
return res;
vector<int> c;
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();++i){
if(i!=0&&nums[i]==nums[i-1])//去重,这里不要忘记i!=0,不然的话会出现数组越界
continue;
for(int j=i+1;j<nums.size();++j){
if(j>i+1&&nums[j]==nums[j-1])//去重,这里不要忘记j>i+1,不然像[0,0,0,0]会通通跳过
continue;
int s=j+1;
int e=nums.size()-1;
while(s<e){
int cur=nums[i]+nums[j]+nums[s]+nums[e];
if(cur==target){
c.clear();
c={nums[i],nums[j],nums[s],nums[e]};
res.push_back(c);
++s;
while(s<e&&nums[s]==nums[s-1])//去重
++s;
}
else if(cur>target){
--e;
while(s<e&&nums[e]==nums[e+1])//去重
--e;
}
else{
++s;
while(s<e&&nums[s]==nums[s-1])
++s;
}
}
}
}
return res;
}
};