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.
分析:
会做3sum就会做4sum,但是这道题目的hash解法我还没来得及研究。
解法:
class Solution {
public:
/**
* @param numbers: Give an array numbersbers of n integer
* @param target: you need to find four elements that's sum of target
* @return: Find all unique quadruplets in the array which gives the sum of
* zero.
*/
vector<vector<int> > fourSum(vector<int> nums, int target) {
// write your code here
vector<vector<int> > res;
if (nums.size() < 4) {
return res;
}
sort(nums.begin(), nums.end() );
for (int i = 0; i < nums.size() - 3; i++) {
if (i != 0 && nums[i] == nums[i - 1]) {
continue;
}
for (int j = i + 1; j < nums.size() - 2; j++) {
if (j != i + 1 && nums[j] == nums[j - 1]) {
continue;
}
int start = j + 1;
int end = nums.size() - 1;
while (start < end) {
int sum = nums[i] + nums[j] + nums[start] + nums[end];
if (sum == target) {
vector<int> quad(4);
quad[0] = nums[i];
quad[1] = nums[j];
quad[2] = nums[start];
quad[3] = nums[end];
res.push_back(quad);
start++;
end--;
while (start < end && nums[start] == nums[start - 1]) {
start++;
}
while (start < end && nums[end] == nums[end + 1]) {
end--;
}
} else if (sum > target) {
end--;
} else {
start++;
}
}
}
}
return res;
}
};