题目:输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
方法一:哈希法(简单但是需要额外空间)
要求a + b = sum, 如果已知a, 那么b = sum - a;所以可以先将b添加入哈希中,然后遍历一遍数组设为a, 在哈希中寻找是否存在sum-a,然后再更新乘积最小值
代码如下:
#include<unordered_map>
class Solution {
public:
//哈希法
vector<int> FindNumbersWithSum(vector<int> array,int sum) {
if(array.empty()){
return vector<int>();
}
int tmp=INT_MAX;
unordered_map<int,int> mp;
pair<int,int>ret;//当需要返回两个数据时可以用pair
for(int i=0;i<array.size();++i){
mp[array[i]]=i;//数组中的数作为哈希表索引,下标作为哈希表的值
}
for(int i=0;i<array.size();++i){
if(mp.find(sum-array[i])!=mp.end()){//即存在和为s的两个数
int j=mp[sum-array[i]];
if(j>i&&array[i]*array[j]<tmp){//寻找乘积最小的两个数
tmp=array[i]*array[j];//更新最小值
ret={i,j};//更新两个数的下标
}
}
}
if(ret.first==ret.second){
return vector<int>();
}
return vector<int>({array[ret.first],array[ret.second]});
}
};
时间复杂度:O(n)
空间复杂度:O(n)
方法二:双指针(更优化,节省空间)
因为数组是有序的,所以可以用双指针,指向数组的首尾,具体步骤如下:
1.初始化:指针i指向数组首, 指针j指向数组尾部
2. 如果arr[i] + arr[j] == sum , 说明是可能解
3. 否则如果arr[i] + arr[j] > sum, 说明和太大,所以–j
4. 否则如果arr[i] + arr[j] < sum, 说明和太小,所以++i
代码如下:
class Solution {
public:
//双指针法(更优化)
vector<int> FindNumbersWithSum(vector<int> array,int sum) {
if(array.empty()){
return vector<int>();
}
int tmp=INT_MAX;
int left=0,right=array.size()-1;
pair<int,int>ret;//当需要返回两个数据时可以用pair
while(left<right){
if(array[left]+array[right]==sum){//即存在和为s的两个数
if(array[left]*array[right]<tmp){//寻找乘积最小的两个数
tmp=array[left]*array[right];//更新最小值
ret={left,right};//更新两个数的下标
}
++left,--right;//当前完成后左指针后移,右指针前移
}else if(array[left]+array[right]<sum){//如果两数之和小于sum,则左指针后移
++left;
}else{//否则,右指针前移
--right;
}
}
if(ret.first==ret.second){
return vector<int>();
}
return vector<int>({array[ret.first],array[ret.second]});
}
};
时间复杂度:O(n)
空间复杂度:O(1)