1 问题
给定一个包含 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]
]
2 解法
此题答案中不可包括重复的数组,因而采用哈希法存在去重困难的问题,使用双指针法。
相比三数之和需要多加一层循环,同时由于target为任意值,当target< 0时,不可添加以下的剪枝条件
//如[-5, -4, 0, 0, 2],target=-9, -5 > -9, 但是-5-4+0+0 = target
if (nums[k] > target)
{
return result;
}
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
//将数组元素从小到大排序
sort(nums.begin(), nums.end());
//定义一个res存放结果
vector<vector<int>> res;
for(int i= 0; i < nums.size(); i++)
{
//去重i位置的元素
if(i > 0 && nums[i] == nums[i-1])
{
continue;
}
for(int j = i + 1; j < nums.size(); j++)
{
//去重j位置的元素
if(j > i + 1 && nums[j] == nums[j - 1])
{
continue;
}
int left = j + 1;
int right = nums.size() - 1;
while(right > left)
{
//四个数和小于target,left右移(扩大和)
if(nums[i] + nums[j] + nums[left] + nums[right] < target)
{
left++;
}
//四个数和大于target,right左移(缩小和)
else if(nums[i] + nums[j] + nums[left] + nums[right] > target)
{
right--;
}
//四个数之和等于target
else
{
//加入结果容器
res.push_back(vector<int>{nums[i], nums[j], nums[left], nums[right]});
//去重left
while(right > left && nums[left] == nums[left + 1])
{
left++;
}
//去重right
while(right > left && nums[right] == nums[right - 1])
{
right--;
}
//收缩left、right寻找下一个
left++;
right--;
}
}
}
}
return res;
}
};