题目
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
思路
- 使用一个哈希表,每遍历一个数字n,计算S-n是否在哈希表中,如果在说明有两个数的和为S。将n添加到哈希表中继续遍历。
- 最后需要返回乘积最小的两个数,每次找到这样两个数的时候比较一下就行了。
代码
class Solution {
public:
vector<int> FindNumbersWithSum(vector<int> array,int sum) {
unordered_map<int, int> hashTable;
vector<int> res(2, 0);
bool flag = false;
int minVal = INT_MAX;
for ( auto& n : array ) {
int diff = sum - n;
if ( hashTable.count( diff ) > 0 ) {
flag = true;
if ( diff*n < minVal ) {
res[0] = diff;
res[1] = n;
minVal = diff*n;
}
}
hashTable[n] = 1;
}
return flag ? res : vector<int>();
}
};