给定一个整型数组a[]和一个数sum,设计一个算法找到数组中所有的整数对(a[i], a[j]),其中a[i] + a[j] == sum。
思路:
可以用bitmap或者哈希来做,这样可以达到O(n)的复杂度。遍历一次数组,将bitmap[a[i]]的值置为1,然后再去检查bitmap[sum - a[i]]的值是否为1,这样就找到满足条件的整数对。如果数组中有重复数字且整数对中的数字不能重复计算,例如:(2, 2, 3),sum = 5,正确结果为{(2, 3)},但bitmap的结果为{(2, 3), (2, 3)},此时可以用哈希表,hash[a[i]]的值为a[i]的出现次数。但是这样的方法可能会造成大量的空间浪费,比如数组的大小较小,但数组中最大值非常大。
另外一个思路就是先排序,然后设置头尾指针往中间扫描,复杂度为O(nlogn)。
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
void FindPair(vector<int> ivec, int sum)
{
if (ivec.size() < 2)
return;
sort(ivec.begin(), ivec.end());
int left = 0, right = ivec.size() - 1;
while (left < right)
{
if (ivec[left] + ivec[right] == sum)
{
cout << ivec[left] << " " << ivec[right] << endl;
++left;
--right;
}
else if (ivec[left] + ivec[right] > sum)
--right;
else
++left;
}
}
int main()
{
vector<int> ivec;
for (int i = 1; i <= 6; ++i)
ivec.push_back(i);
int sum = 5;
FindPair(ivec, sum);
return 0;
}