昨天面试了一家大厂,第一道面试题是面试官叫我两分钟写一个二分查找算法,我的代码是这样的:
package array;
public class _二分查找 {
public static void main(String[] args) {
int [] arr = {1,2,3,4,5,6,6,100};
int var = 2;
System.out.println(fun(arr,var));
}
public static int fun(int [] arr,int var) {
int left = 0;
int right = arr.length-1;
while(left < right) {
int mid = (left + right)/2;
if(var == arr[mid]) {
return mid;
}else if(var > arr[mid]) {
left = mid+1;
}else {
right = mid-1;
}
}
return -1;
}
}
看起来好像没什么问题,面试官又问,请问还可以怎么改进吗?当时我就懵了,然后我想了想,left+right可能会越界。。。。。。面试官继续问,除了这个呢?我想了好几分钟,没想出来,今天来复盘的时候发现,当var小于arr[left]或者var大于arr[rigth]的时候,我们可以直接返回,不用执行下面的代码了,改进以后是这样的:
package array;
public class _二分查找 {
public static void main(String[] args) {
int [] arr = {1,2,3,4,5,6,6,100};
int var = 2;
System.out.println(fun(arr,var));
}
public static int fun(int [] arr,int var) {
int left = 0;
int right = arr.length-1;
if(var < arr[left] || var > arr[right] || left > right){
return -1;
}
while(left < right) {
int mid = (left + right)/2;
if(var == arr[mid]) {
return mid;
}else if(var > arr[mid]) {
left = mid+1;
}else {
right = mid-1;
}
}
return -1;
}
}
这样应该就是面试官想要的答案了。