【题目描述】
对于一个有序数组,我们通常采用二分查找的方式来定位某一元素,请编写二分查找的算法,在数组中查找指定元素。
给定一个整数数组A及它的大小n,同时给定要查找的元素val,请返回它在数组中的位置(从0开始),若不存在该元素,返回-1。若该元素出现多次,请返回第一次出现的位置。
【测试样例】
[1,3,5,7,9],5,3
【返回结果】
1
【分析】
3个需要注意的题目问题:
(1)有序数组;
(2)二分查找;
(3)若元素出现多次,要返回第一次出现的位置。
【Solultion】
class BinarySearch {
public:
int getPos(vector<int> A, int n, int val) {
if (n == 0) return -1;
if (n == 1) return 0;
int high = n - 1;
int low = 0;
while (low < high) {
int mid = low + (high - low)/2; //防止溢出
if (val == A[mid]) {
high = mid; //如果找到了val,则继续对low~mid段数据进行二分查找
} else if (val > A[mid]) {
low = mid + 1;
} else {
high = mid - 1;
}
}
if (A[high] == val) { //如果存在val,则high对应的值就是第一次出现的位置
return high;
}
return -1;
}
};