描述
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]
]
给定一个数组以及一个目标值,找出所有四元组,它们的和为目标值。
分析
参考 Two Sum 以及 3Sum 这两个问题,其实后一个问题是在前一个问题的基础上加上一层循环,对应的这个问题也是在“3Sum”这个问题的基础上再加一层循环。
建议大家把相关的这几道题目放在一起研究。
代码
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
set<vector<int>> res;
sort(nums.begin(), nums.end());
for (int l = 0; l < nums.size(); l++) {
for (int k = l + 1; k < nums.size(); k++) {
int i = k + 1, j = nums.size() - 1, sum = target - nums[l] - nums[k];
while (i < j) {
if (nums[i] + nums[j] == sum)
res.insert({nums[l], nums[k], nums[i++], nums[j--]});
if (nums[i] + nums[j] < sum) i++;
else if (nums[i] + nums[j] > sum) j--;
}
}
}
return vector<vector<int> > (res.begin(), res.end());
}
};