题意:给定一个包含 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] ]
思路:与三数之和类似,相当于在外面再套一层循环
主要注意各部分的剪枝与去重,一步一步做,和三数之和类似
在一级剪枝时,判断条件要注意,nums[i]>0且target>0
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();i++)
{
//一级剪枝
if(nums[i]>target&&nums[i]>0&&target>=0)
break;
else
{
//对i去重
//注意这里i>0一定要写在前面,否则会出错
if(i>0&&nums[i]==nums[i-1])
continue;
else
{
for(int j=i+1;j<nums.size();j++)
{
//二级剪枝
if(nums[j]>0&&target>0&&nums[i]+nums[j]>target)
break;
else
{
//对j去重
if(nums[j]==nums[j-1]&&j>i+1)
continue;
else
{
int left=j+1;
int right=nums.size()-1;
while(right>left)
{
if((long)nums[i]+nums[j]+nums[left]+nums[right]>target)
right--;
else if((long)nums[i]+nums[j]+nums[left]+nums[right]<target)
left++;
else
{
result.push_back(vector<int>{nums[i],nums[j],nums[left],nums[right]});
//对left、right去重
while(right>left&&nums[right]==nums[right-1])
{
right--;
}
while(right>left&&nums[left]==nums[left+1])
{
left++;
}
right--;
left++;
}
}
}
}
}
}
}
}
return result;
}
};