leetcode 4Sum
- 题目描述:
- 给定一个包含n个整数的数组S和目标整数target,在S中找4个整数a,b,c,d,使得这a + b + c + d = target。找出所有符合这种情况的四元组。最终结果中的四元组要唯一且四元组内元素非降序。
- 题目来源:
- http://oj.leetcode.com/problems/4sum/
- 题目分析:
- 先对数组排序,然后构造键为某个和sum,值为所有符合和为sum的元素对的map结构。枚举第一和第二个元素,通过target值得到剩余两个元素的和sum1,使用map中的和为sum1的对象构造第三和四个元素。
- 正确性说明,假设结果集中某个四元组为a,b,c,d,枚举a,b,通过map中键为target - a - b的值中可以找到c,d,所以解一定在结果集中,剩下的是去掉重复的四元组。使用另一个map对象记录计算过程中已经出现的a,b,这样保证元素对a,b不会重复计算。在对原数组排过序的情况下,如果存在多个相同的元素对,可以保证值相同的元素对在上面介绍的第一个map中的位置相邻,那么可以通过和在结果集中的最近添加的四元组比较来保证元素对c,d不重复(a,b元素对确定的情况下)。
- 示例代码:
vector<vector<int> > fourSum(vector<int> &num, int target) { vector<vector<int> > ans; sort(num.begin(),num.end()); map<int, vector<pair<int,int> > > m; map<pair<int,int>, int> sign; int anssize = 0; int numsize = num.size(); for(int i = 0; i < numsize; ++i) { for(int j = i + 1; j < numsize; ++j) { m[num[i] + num[j]].push_back(pair<int, int>(i,j)); } } for(int i = 0; i < numsize; ++i) { for(int j = i + 1; j < numsize; ++j) { if(sign.find(pair<int, int>(num[i], num[j])) != sign.end()) continue; sign[pair<int, int>(num[i], num[j])] = 1; int x = (target - num[i] - num[j]); if(m.find(x) != m.end()) { for(vector<pair<int,int> >::iterator it = m[x].begin(); it != m[x].end(); ++it) { if(it->first <= j) { continue; } if(anssize == 0 || ans[anssize - 1][0] != num[i] || ans[anssize - 1][1] != num[j] || ans[anssize - 1][2] != num[it->first] ) { vector<int> tmp; tmp.push_back(num[i]); tmp.push_back(num[j]); tmp.push_back(num[it->first]); tmp.push_back(num[it->second]); ans.push_back(tmp); ++anssize; } } } } } return ans; }