前言
上一篇博客我们已经讲了两数之和最常规的解法:
博客地址:两数之和
这篇博客讲讲还有哪些方法。
思路
对于任意一个数组,我们可以根据给定的sum计算出一个对于的数组:
例如在排序后为:1,3,5,7,9,15,sum为10,我们可以让sum减去每个元素得到新的数组为:9,7,5,3,1,-5.我们可以通过两个指针i与j,规定i < j,如果 i > j,让i从原数组左边开始移动,让j从右数组右边开始移动,当i == j时就说明找到了,直接跳出循环,当
array[i] 等于 array[j] 时说明找到了;
array[i] < array[j]时,i++;
array[i] > array[j]时,j++;
话不多说,附上代码!
package interview;
public class TwoSum2 {
public static void main(String[] args) {
//定义一个数组
int [] array = {3,1,7,5,9,15};
quickSort(array,0,array.length-1);
twoSum(array,10);
// print(array);
}
public static void twoSum(int [] array,int sum) {
int [] array2 = new int[array.length];
for (int i = 0; i < array.length; i++) {
array2[i] = sum - array[i];
}
int i = 0;
int j = array.length-1;
while (i < j) {
if (array[i] == array2[j]) {
System.out.println("两个数分别为:"+array[i]+" "+array[j]);
return;
}else if (array[i] < array2[j]){
while (i < j && array[i] < array2[j]) {
i++;
}
}else {
while (i < j && array[i] > array2[j]) {
j--;
}
}
}
}
//快速排序
public static void quickSort(int [] array,int begin,int end) {
if (begin < end) {
int i = begin;
int j = end;
int key = array[i];
if (i < j) {
while (key < array[j]) {
j--;
}
if(i < j) {
array[i] = array[j];
i++;
}
while (i < j && key > array[i]) {
i++;
}
if(i < j) {
array[j] = array[i];
j--;
}
array[i] = key;
quickSort(array,begin,i-1);
quickSort(array,i+1,end);
}
}
}
//打印数组
public static void print(int[] array) {
for (int i = 0; i < array.length; i++) {
System.out.println(array[i]);
}
}
}
这种方法的时间复杂度为O(n),当然是排除了排序的时间消耗,但空间复杂度并不是太好,也为O(n),下面再做一点优化:
我们不再赋值数组,就用原来的数组做判断,每次都计算两个下标位置的值的和,看看是否等于sum。
附上代码:
public static void twoSum(int [] array,int sum) {
int [] array2 = new int[array.length];
for (int i = 0; i < array.length; i++) {
array2[i] = sum - array[i];
}
int i = 0;
int j = array.length-1;
while (i < j) {
int sumTemp = array[i] + array[j];
if (sumTemp == sum) {
System.out.println("两个数分别为:"+array[i]+" "+array[j]);
return;
}else if (sumTemp > sum){
j--;
}else {
i++;
}
}
}
(由于只对twoSum()方法进行了改动所以只附上了部分代码)
我们对第一种方法进行了优化,只是少用了一个数组,这样就达成了时间复杂度为O(n),空间复杂度为O(1)的算法实现。
总结
遇到问题一定要多思考要没有更好的方法,不仅可以锻炼我们的编码能力,还能让我们的思维得到了提升。