四数之和
题目描述:
给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。
注意:
答案中不可以包含重复的四元组。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
int left,right;
vector<vector<int>> result;
if(nums.size() <= 3) return result;
set<vector<int>> s ; // 利用set容器进一步去重
sort(nums.begin(),nums.end()); // 排序
for(int i = 0 ; i<nums.size()-2 ; i++){
for(int j = i + 1 ;j<nums.size()-1 ; j++){
left = j + 1;
right = nums.size() - 1;
while(left < right){
int tempSum = nums[i] + nums[j] + nums[left] + nums[right] - target;
if(tempSum == 0){
int temp[] = {nums[i],nums[j],nums[left],nums[right]};
vector<int> v(temp,temp+4);
s.insert(v);
left ++;
right --;
}else if(tempSum > 0){
right --;
}else{
left ++;
}
}
}
}
for(set<vector<int>>::iterator it = s.begin();it != s.end() ; it++){
result.push_back(*it);
}
return result;
}
};
和三数之和思路类似,在三数之和的基础上多一重for循环。