题目
输入一个递增排序的数组和一个数字S,在数组中查找两个数,是的他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
输出:对应每个测试案例,输出两个数,小的先输出。
思路
设置头尾指针,每次把两个指针指向的元素相加,如果和为sum则输出这两个元素,如果小于sum则把头指针右移,如果大于sum则把尾指针左移。由于是要求乘积最小的,所以这两个元素差的越远则越小,即第一次找到的符合要求的两个数就是乘积最小的。
参考代码
class Solution {
public:
vector<int> FindNumbersWithSum(vector<int> array, int sum) {
vector<int> res(2);
res[0] = res[1] = -1;
int length = array.size();
int i = 0, j = length - 1;
while (i < j) {
if (array[i] + array[j] > sum) {
j--;
} else if (array[i] + array[j] < sum) {
i++;
} else {
res[0] = array[i];
res[1] = array[j];
break;
}
}
return res[0] == -1 ? vector<int>() : res;
}
};