二分查找
题目描述
对于一个有序数组,我们通常采用二分查找的方式来定位某一元素,请编写二分查找的算法,在数组中查找指定元素。
给定一个整数数组A及它的大小n,同时给定要查找的元素val,请返回它在数组中的位置(从0开始),若不存在该元素,返回-1。若该元素出现多次,请返回第一次出现的位置。
解题思路
与普通的二分查找相比,需注意元素多次出现。
代码
import java.util.*;
public class BinarySearch {
public int getPos(int[] A, int n, int val) {
// write code here
int lo = 0;
int hi = n-1;
while(hi>=lo){
int mid = lo + (hi - lo) / 2;
if (val < A[mid]) hi = mid - 1;
else if (val > A[mid]) lo = mid + 1;
else {
// 该if用于解决相邻元素相同的情况
if (mid > 0 && A[mid-1] == A[mid]) return mid-1;
else return mid;
}
}
return -1;
}
}