难度:中等
给定一个包含 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] ]
题目解析:
原理是与“三数之和”是一样的,还是使用双指针法。在“三数之和”的基础上,加上一个for循环。这里有个问题,就是前面两个指针是不能一起移动的,不然就像“三数之和”中的错误解法,两个指针固定,一个指针找,然后两个指针一起移动,就会遗漏。
参考代码:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int> > res;
if (nums.size() < 4)
return res;
vector<int> temp;
int size = (int)nums.size();
int left = 2;
int right = size - 1;
int j;
sort(nums.begin(),nums.end());
for (int i = 0; i <= nums.size()-4; i++)
{
for ( j = i+1; j <= nums.size()-3; j++)
{
left = j + 1;
right = size - 1;
while (left < right)
{
int l_num = nums[left];
int r_num = nums[right];
if ((nums[i] + nums[j] + nums[left] + nums[right]) == target)
{
temp.push_back(nums[i]);
temp.push_back(nums[j]);
temp.push_back(nums[left]);
temp.push_back(nums[right]);
res.push_back(temp);
temp.clear();
while (left < right && nums[left] == l_num) ++left;
while (left < right && nums[right] == r_num) --right;
}
else if ((nums[i] + nums[j] + nums[left] + nums[right]) < target)
{
++left;
}
else if ((nums[i] + nums[j] + nums[left] + nums[right]) > target)
{
--right;
}
}
while ((j+1) < nums.size() && nums[j] == nums[j + 1])
{
j++;
}
}
while (i < j && (i + 1) < nums.size() && nums[i] == nums[i + 1])
{
i++;
}
}
return res;
}
};