三种二分模板,>=,>,=。
<的话,=下标减一就好了。
最后一个=的话,>下标减一就好了。
注意:这里的mid不要用(l+r)/2,因为可能会溢出。
//第一个 大于等于
int findFirstBigAndEqual(vector<int> a, int num)
{
int n = a.size();
int l = 0;
int r = n - 1;
while (l <= r)
{
int mid = l+(r - l) / 2;
if (a[mid] >= num)
r = mid - 1;
else
l = mid + 1;
}
return l >= n ? -1: l;
}
//第一个 大于
int findFirstBig(vector<int> a, int num)
{
int n = a.size();
int l = 0;
int r = n - 1;
while (l <= r)
{
int mid = l + (r - l) / 2;
if (a[mid] > num)
r = mid - 1;
else
l = mid + 1;
}
return l >= n ? -1 : l;
}
//等于
int findEqual(vector<int> a, int num)
{
int n = a.size();
int l = 0;
int r = n - 1;
while (l <= r)
{
int mid = l + (r - l) / 2;
if (a[mid] > num)
r = mid - 1;
else if (a[mid] == num)
return num;
else
l = mid + 1;
}
return -1;
}
STL里面的模板
ForwardIter lower_bound(ForwardIter first, ForwardIter last,const _Tp& val)算法返回一个非递减序列[first, last)中的第一个大于等于值val的位置。
//这个算法中,first是最终要返回的位置
int lower_bound(int *array, int size, int key)
{
int first = 0, middle;
int half, len;
len = size;
while(len > 0) {
half = len >> 1;
middle = first + half;
if(array[middle] < key) {
first = middle + 1;
len = len-half-1; //在右边子序列中查找
}
else
len = half; //在左边子序列(包含middle)中查找
}
return first;
}
ForwardIter upper_bound(ForwardIter first, ForwardIter last, const _Tp& val)算法返回一个非递减序列[first, last)中第一个大于val的位置。
int upper_bound(int *array, int size, int key)
{
int first = 0, len = size-1;
int half, middle;
while(len > 0){
half = len >> 1;
middle = first + half;
if(array[middle] > key) //中位数大于key,在包含last的左半边序列中查找。
len = half;
else{
first = middle + 1; //中位数小于等于key,在右半边序列中查找。
len = len - half - 1;
}
}
return first;
}