原题链接:和为S的两个数字_牛客题霸_牛客网
描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,返回两个数的乘积最小的,如果无法找出这样的数字,返回一个空数组即可。
返回值描述:
对应每个测试案例,输出两个数,小的先输出。
示例1
输入:
[1,2,4,7,11,15],15
复制返回值:
[4,11]
解题思路:
这道题的考点应该在于递增这两个字。既然数组是递增的,那么我就一定要用到二叉树查找的概念。输入一个数S,那我们从头开始计算,S=15,先读到1,15-1=14,直接二叉树查找是否存在14。同理,接着找15-2=13是否存在,这样循环查找下去。
另外这里数组存在下面这种特殊情况要考虑,两个相同的值相加等于S。
[1,4,4,9],8。
代码:
public ArrayList<Integer> FindNumbersWithSum(int[] array, int sum) {
ArrayList<Integer> list = new ArrayList<>();
for (int i = 0; i < array.length; i++) {
int key = array[i];
int findValue = sum - key;
if (findValue == key) {
//如果查找值等于当前值,比如查找值等于4,S=8的情况。则判断4的前后是否有4
if ((i > 0 && array[i - 1] == findValue) || (array[i + 1] == findValue && i < (array.length - 2))) {
list.add(key);
list.add(findValue);
return list;
}
continue;
}
boolean b = binaryLookUp(array, findValue);
if (b) {
list.add(key);
list.add(findValue);
return list;
}
}
return list;
}
/**
* 二叉树查找
*
* @param array
* @param findValue
* @return 是否存在
*/
private boolean binaryLookUp(int[] array, int findValue) {
int start = 0;
int end = array.length - 1;
while (true) {
int index = (start + end) / 2;
int value = array[index];
if (value == findValue) {
return true;
}
if (value > findValue) {
end = end - (end - start) / 2;
} else {
start = start + (end - start) / 2;
}
if (start == end) {
return false;
}
if ((end - start) == 1) {
return array[start] == findValue || array[end] == findValue;
}
}
}