您的位置:首页 > 其它

4Sum

2015-10-29 21:13 288 查看

题目: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)

思路:

先是保存前面数字两两之和,接下来,和之前一样,上来就是两个循环,如果有重复的,continue。

后面就是看是不是有target-num[i]-num[j],有的话,则开始最重要的程序了,首先,看hash里面的第一个数值是否大于刚刚的j,因为我们是从后往前的。不可能小于。

还有就是要区分是否是第一次放进去,如果第二次放进去的话,那么,就需要考虑 res中倒数第二个数,不能够和hash表第一个数相同。

因为此时,第一个,第二个数都是相同的,第三个数相同,那么第四个一定相同。

代码:

class Solution {
public:
//http://www.cnblogs.com/tenosdoit/p/3649607.html 非常好的一篇博客
vector<vector<int> > fourSum(vector<int> &num, int target) {
int n=num.size();
vector<vector<int> > res;
unordered_map<int,vector<pair<int,int> >>pairs;
sort(num.begin(),num.end());
//先保存 前面两数的和
for(int i=0;i<=n-2;i++){
for(int j=i+1;j<=n-1;j++){
pairs[num[i]+num[j]].push_back(make_pair(i,j));
}
}

for(int i=0;i<=n-4;i++){
if(i>0&&num[i]==num[i-1]) continue;//避免重复元素
for(int j=i+1;j<=n-3;j++){
if(j>i+1&&num[j]==num[j-1]) continue;

if(pairs.find(target-num[i]-num[j])!=pairs.end()){
bool isFirstPush=true;
vector< pair<int,int> > sum2=pairs[target-num[i]-num[j]];
for(int k=0;k<sum2.size();k++){
if(sum2[k].first<=j) continue;//新找出来的数组索引值一定要大于之前的两个索引值,i,j
if(isFirstPush||(res.back())[2]!=num[sum2[k].first]){
res.push_back(vector<int>{num[i],num[j],num[sum2[k].first],num[sum2[k].second]});
isFirstPush=false;
//这里较难理解,因为我第一次可以有重复 1 2 2 3,但是第二次就不行
//如果有1 2 2,第四个数字一定是3,就一定会重复
}
}
}
}
}

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