力扣上求四数之和的题,我想把它变成求n数之和,但是不知道为什么会出现错误:
代码如下:
class Solution {
public:
set<vector<int>> ans;
vector<vector<int>> fourSum(vector<int>& nums, int target) {
int len =nums.size();
if(len < 4){
return setToVector(ans);
}
vector<int> temp = vector<int>();
sort(nums.begin(),nums.end());//排序
Nsum(nums,target,4,temp);
return setToVector(ans);
}
void Nsum(vector<int> nums,int target,int n,vector<int> temp){
if(n == 1 ){
for(auto i = nums.begin();i != nums.end();i++){
if(*i == target){
temp.push_back(target);
sort(temp.begin(),temp.end());//重新排序
ans.insert(temp);
temp.pop_back();
}
}
}else{
for(int i = 0;i < nums.size();i++){
int c = nums[i];
temp.push_back(c);
nums.erase(nums.begin()+i);//返回值是一个迭代器
Nsum(nums,target - c,n-1,temp);
temp.pop_back();//删除刚刚添加的元素
nums.insert(nums.begin()+i,c);
}
}
}
vector<vector<int>> setToVector(set<vector<int>> ans){
vector<vector<int>> result;
if(ans.size()==0){
return result;
}
for(auto i=ans.begin(); i!=ans.end();i++){
result.push_back(*i);
}
return result;
}
};
问题:为什么会出现四数和不为target的组合: