[leetcode] 18. 4Sum

18. 4Sum

Given an array S of n integers, are there elements a, b, c, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.

Note: The solution set must not contain duplicate quadruplets.

For example, given array S = [1, 0, -1, 0, -2, 2], and target = 0.

A solution set is:

[

[-1, 0, 0, 1],

[-2, -1, 1, 2],

[-2, 0, 0, 2]

]

题意：

给定一个含有n个整数的数组，在给定的数组中，找到四个数，使其之和等于目标值。在给定数组中找到所有的等于目标值的不相同的四元组。

注意：答案要求四元组集合不包含重复的元素。

思路：

这道题的思路和之前的 [leetcode] 15. 3Sum 差不多，首先对数组排序，确定四元数中的前两个（a，b）， 遍历剩余数组确定两外两个（c，d），确定cd时思路跟 [leetcode] 15. 3Sum 确定后两个数据一样，二分查找左右逼近，在去重时采用set集合。用两个指针pre和tail从b这个数后面的两边往中间查找，用第一重for循环确定第一个数字，用第二重for循环确定第二个数字，然后在while循环里，第三、第四个数字分别从两端往中间扫。当这四个数之和等于target的时候，记录一下，当之和大于target的时候，tail左移，而当之和小于target的时候，pre右移，直到pre和tail相遇。这个其实就是在3Sum外层加1层循环。 代码如下：

class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
set<vector<int>> res2;//去重
vector<vector<int>> res;
sort(nums.begin(), nums.end());
int len = nums.size();
for (int i = 0;i<len-3;i++)
{
for (int j = i+1;j<len-2;j++)
{
int pre = j + 1;
int tail = len - 1;
while (pre < tail)
{
vector<int> tempRes;
int sum = nums[i] + nums[j] + nums[pre] + nums[tail];
if (sum == target)
{
tempRes.push_back(nums[i]);
tempRes.push_back(nums[j]);
tempRes.push_back(nums[pre]);
tempRes.push_back(nums[tail]);
res2.insert(tempRes);
pre++;
tail--;
}
else if (sum < target)
{
pre++;
}
else
tail--;
}
}
}
for (auto r : res2)
{
res.push_back(r);
}
return res;
}
};