58 四数之和
原题网址:https://www.lintcode.com/problem/4sum/description
描述
给一个包含n个数的整数数组S,在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。
四元组(a, b, c, d)中,需要满足a <= b <= c <= d
答案中不可以包含重复的四元组。
您在真实的面试中是否遇到过这个题?
样例
例如,对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为:
(-1, 0, 0, 1)
(-2, -1, 1, 2)
(-2, 0, 0, 2)
标签
排序
哈希表
两根指针
数组
思路:沿用三数之和的方法,对数组排序,然后固定两个数,再设置两个指针左右滑动,时间复杂度为O(n^3)。注意每个索引都要去除重复。
AC代码:
class Solution {
public:
/**
* @param numbers: Give an array
* @param target: An integer
* @return: Find all unique quadruplets in the array which gives the sum of zero
*/
vector<vector<int>> fourSum(vector<int> &numbers, int target) {
// write your code here
vector<vector<int>> result;
int n=numbers.size();
sort(numbers.begin(),numbers.end());
for (int i=0;i<n;i++)
{
if (i>0&&numbers[i]==numbers[i-1])
{
continue;
}
for (int j=i+1;j<n;j++)
{
if (j>i+1&&numbers[j]==numbers[j-1])
{
continue;
}
int left=j+1,right=n-1;
while(left<right)
{
if (left>j+1&&numbers[left]==numbers[left-1])
{
left++;
continue;
}
if (right<n-1&&numbers[right]==numbers[right+1])
{
right--;
continue;
}
int sum=numbers[i]+numbers[j]+numbers[left]+numbers[right];
if (sum==target)
{
vector<int> temp(4,numbers[i]);
temp[1]=numbers[j];
temp[2]=numbers[left];
temp[3]=numbers[right];
result.push_back(temp);
left++;
right--;
}
else if (sum<target)
{
left++;
}
else
{
right--;
}
}
}
}
return result;
}
};
去除重复除了使用if——continue的方式,还可以用while循环:
while(left < right && numbers[left] == numbers[left+1])
{
left++;
}
while(left < right && numbe [right] == numbers[right-1])
{
righ--;
}
此外,还可以用set,参考:http://www.cnblogs.com/grandyang/p/4515925.html
其他方法:哈希表,通过增加空间复杂度来降低时间复杂度:https://www.cnblogs.com/eudiwffe/p/6282635.html