您的位置:首页 > 其它

[LeetCode18]4Sum

2015-11-09 13:54 323 查看
题目来源:https://leetcode.com/problems/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:

Elements in a quadruplet (a,b,c,d) must be in non-descending order. (ie, a ≤ b ≤ c ≤ d)
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)

题目分析:
这道题和3Sum几乎同出一辙,
只不过是要求四个数字的和, 在时间复杂度上要比3Sum高一个数量级。

于两点要求的处理:
1.
首先要对整个数组进行排序,
这样得到的答案自然是排序好的.
2.
对于重复答案的处理和3Sum是一摸一样的。
解题思路:
先排序,然后左右夹逼,复杂度
O(n3),会超时。
可以用一个
hashmap 先缓存两个数的和,最终复杂度
O(n3)。这个策略也适用于
3Sum


时间复杂度分析:
这道题的解法, 我选择的是空间复杂度为1,
时间复杂度为O(n3).对于这样的问题,
如果到了KSum(K>=5),
我觉得可以用hash_map来牺牲空间复杂度换取好一些的时间复杂度.

class Solution18{
public:
vector<vector<int>>fourSum(vector<int> &nums, int target){
vector<vector<int>> result;
if (nums.size() < 4) return result;

sort(nums.begin(), nums.end());

//auto last = nums.end();
for (auto i = nums.begin(); i < prev(nums.end(), 3); ++i){
for (auto j = next(i); j < prev(nums.end(), 2); ++j){
auto k = next(j);
auto l = prev(nums.end());
while (k < l){
if (*i + *j + *k + *l < target){
++k;
}
else if (*i + *j + *k + *l > target){
--l;
}
else{
result.push_back({ *i, *j, *k, *l });
++k;
--l;
}

}
}
}
sort(result.begin(), result.end());
result.erase(unique(result.begin(), result.end()), result.end());
for (auto i = 0;i < result.size(); ++i){
for (auto j = 0; j < result[i].size(); ++j){

cout << result[i][j] << " ";
}
cout << endl;
}

return result;

}
};

int main(){
Solution18 solution;
vector<int> vec = { 1, 0, -1, 0, -2, 2 };
int target = 0;
solution.fourSum(vec, target);

getchar();
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: