剑指offer(四十二)——和为S的两个数
题目描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
题解
这道题一开始被误导了,想通了就比较简单了。这里面给出了两个限制条件来查找目标元素:(1)和为S;(2)乘积最小。仔细想想,在和相同的情况下,两数差值越大,积不就越大?问题就迎刃而解了,数组是有序的,那么首尾两个元素一定是差值最大的,从这两个元素开始进行遍历,找到第一对符合条件的元素就是目标元素了。
这里画个图给大家看看吧:
- 假设sum=155,我们可以人工算得array[1] + array[6] = 155,array[2] + array[5] = 155。显然是array[1]和array[6]的乘积更小,符合条件。
- 定义两个指针,一首一尾,用来定位两个元素的位置,通过指针移动寻找目标元素。
- 计算两个指针所在的元素值之和,若大于sum,则post指针向左移动,若小于sum,则pre指针向右移动。
- 直至找到目标元素或pre指针不小于post指针。
- sum = array[0] + array[8] = 180 > sum,移动post指针
- sum = array[0] + array[7] = 157 > sum,移动post指针
- sum = array[0] + array[6] = 144< sum,移动pre指针
- sum = array[1 + array[6] = 150=sum,找到第一对目标元素,存入list,跳出循环,返回结果。
public ArrayList<Integer> FindNumbersWithSum(int [] array,int sum) {
int post = array.length - 1;
ArrayList<Integer> anslist = new ArrayList<>();
int pre = 0;
if (array.length < 2) {
return anslist;
}
while(post > pre) {
int tempsum = array[post] + array[pre];
if (tempsum < sum) {
pre++;
}else if (tempsum > sum) {
post--;
}else {
anslist.add(array[pre]);
anslist.add(array[post]);
break;
}
}
return anslist;
}