题目:输入一个递增排序的数组和一个数字s,在数组中查找两个数,使得它们的和正好是s。如果有多对的和等于s,则输出全部对。
思路
例如输入数组{1,2,4,7,11,15}和数字15.由于4+11=15,因此输出4和11.
最简单肯定是O(n*n)
的方法,也就是先在数组中固定一个数字,再依次判断数组中其余n-1个数字与它的和是不是等于s。时间复杂度太高,效率太低。
接着我们寻找更好的算法。我们先在数组中选择两个数字,如果他们的和等于输入的s,我们就找到了要找的两个数字。如果小于s呢?我们希望两个数字的和再大一点。由于数组已经排好序了,我们可以考虑选择较小的数字后面的数字。因为排在后面的数字要大一点,那么这两个数字的和也要大一点,就有可能等于输入的数字s了。同样,当两个数字的和大于输入的数字的时候,我们就可以选择较大的数字前面的数字了,因为排在前面的数字要小一些。
package com.zhumq.leetcode;
import org.junit.Test;
public class FindSumTwoNums {
/*
* 先选择数组尾首两个数,下标分别为small=0,big=arr.length-1
* 如果和等于输入的s,我们就找到了要找的两个数字。
* 如果小于s,由于数组已经排好序了,选择较小的数字后面的数字,即small++。
* 同理当两个数字的和大于输入的数字的时候,选择较大的数字前面的数字了,即big--。
*/
public void findSumTwoNums(int arr[],int target) {
if(arr==null||arr.length<2) return;
int small = 0;
int big = arr.length-1;
int sum = 0;
while(small<big) {
sum = arr[small]+arr[big];
if(sum == target) {
System.out.println(arr[small]+" "+arr[big]);
//当可能存在多组结果时,接着遍历直到small=big
small++;
big--;
}else if(sum>target) {
big--;
}else {
small++;
}
}
}
@Test
public void test1() {
int arr[] = {9,10,11,12,13};
findSumTwoNums(arr, 23);
}
}