LintCode 14. 二分查找
问题描述
给定一个排序的整数数组(升序)和一个要查找的整数target,用O(logn)的时间查找到target第一次出现的下标(从0开始),如果target不存在于数组中,返回-1。
样例
在数组 [1, 2, 3, 3, 4, 5, 10] 中二分查找3,返回2。
问题分析
要求用二分查找,二分查找的思路是将表中间位置记录的关键字与查找关键字比较,如果两者相等,则查找成功;否则利用中间位置记录将表分成前、后两个子表,如果中间位置记录的关键字大于查找关键字,则进一步查找前一子表,否则进一步查找后一子表。重复以上过程,直到找到满足条件的记录,使查找成功,或直到子表不存在为止,此时查找不成功。特别注意的是,这个问题要求返回目标第一次出现的位置,所以还要做一下改进,当中间位置的关键字和查找的关键字相同时,还有判断中间关键字之前有没有同样大小的关键字。
代码
class Solution {
public:
/**
* @param nums: The integer array.
* @param target: Target number to find.
* @return: The first position of target. Position starts from 0.
*/
int binarySearch(vector<int> &array, int target) {
// write your code here
int low = 0;
int high = array.size();
int mid,pos = -1;
while(low <= high)
{
mid = (low + high) / 2;
if (array[mid] == target) {
pos = mid;
}
if (array[mid] >= target) {
high = mid - 1;
} else {
low = mid + 1;
}
}
return pos;
}
};