题目描述 [和为S的两个数字]
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
对应每个测试案例,输出两个数,小的先输出。
解题思路
设置两个指针,i,j分别表示数组的开始和末尾位置,i=1,j=array.size()-1;;设置一个临时变量product存储两个数的乘积,设置nums表示返回的两个数;
- 如果i,j下标对应数字之和大于sum, j--;
- 如果i,j下标对应数字之和小于sum, i++;
- 如果i,j下标对应数字之和等于sum,判断i,j下标对应数字的乘积是否小于临时变量,如果小于则更新nums和product,否则不更新。
代码
class Solution {
public:
vector<int> FindNumbersWithSum(vector<int> array,int sum) {
vector<int> nums;
int product = INT_MAX;
int i=0;
int j=array.size()-1;
while(i<j){
int sum_ij = array[i] + array[j];
if(sum_ij==sum){
if(array[i]*array[j]<product){
nums.clear();
nums.push_back(array[i]);
nums.push_back(array[j]);
product = array[i]*array[j];
}
i++;
j--;
}else if(sum_ij>sum){
j--;
}else{
i++;
}
}
return nums;
}
};