和为S的两个数字
描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。对应每个测试案例,输出两个数,小的先输出。
代码 (Java)
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> FindNumbersWithSum(int [] array, int sum) {
ArrayList<Integer> result = new ArrayList<>();
if (array == null || array.length < 2)
return result;
int minProduct = Integer.MAX_VALUE;
boolean found = false;
int num1 = 0, num2 = 0;
int small = 0, big = array.length - 1;
while (small < big) {
int curSum = array[small] + array[big];
if (curSum == sum) {
found = true;
int product = array[small] * array[big];
if (product < minProduct) {
minProduct = product;
num1 = array[small];
num2 = array[big];
}
big--;
small++;
} else if (curSum > sum)
big--;
else
small++;
}
if (found) {
result.add(num1);
result.add(num2);
return result;
}
return result;
}
}
思路
- 本题与 和为S的连续正数序列 类似,这题也用两个指针,从数组的第一个和最后一个数字开始,往中间遍历,详细看代码就可以了,时间复杂度是O(n)。