一、题目描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
输出描述:
对应每个测试案例,输出两个数,小的先输出。
二、思路
- 给定数组是有序的,因此可以考虑首尾指针。i指向array第一个元素,j指向最后一个元素。
- 当array[i]+array[j] > S 时,j往前挪一位;当array[i]+array[j] < S 时,i往后挪一位;相等则比较记录。时间复杂度O(n)。
- 可优化:i,j可以用二分查找来更新。
- 如果给定数组无序的话,可以考虑用map,map.get(S-a[i]),空间换时间。
三、代码
public ArrayList<Integer> FindNumbersWithSum(int [] array,int sum) {
int len = array.length;
ArrayList<Integer> ans = new ArrayList<>();
if (array == null || len == 0) return ans;
if (array.length == 1) {
ans.add(array[0]);
return ans;
}
int mn = 0x7fffffff;
for (int i=0, j=len-1; i<len && j>i;) {
if (array[i] + array[j] == sum && array[i] * array[j] < mn) {
mn = array[i] * array[j];
ans.clear();
ans.add(array[i]);
ans.add(array[j]);
i++;
} else if (array[i] + array[j] > sum) {
j--;
} else {
i++;
}
}
return ans;
}