关于区间的选择,觉得还是闭区间优于左闭右开,选用闭区间。
//[l, r]
//相等的,找不到返回-1
int bs1(int *d, int l, int r, int v){
while(l <= r){
int m = (l+r)>>1;
// printf("%d %d %d\n", l, r, m);
if(v > d[m]){
l = m + 1;
}else if(v < d[m]){
r = m - 1;
}else{
return m;
}
}
return -1;
}
//第一个相等的,不存在则返回第一个>v的,即第一个>=v
int bs2(int *d, int l, int r, int v){
while(l <= r){
int m = (l+r)>>1;
// printf("%d %d %d\n", l, r, m);
if(v > d[m]){
l = m + 1;
}else{
r = m - 1;
}
}
return l;
}
//最后一个相等的,不存在则返回最后一个<v的,即最后一个<=v
int bs3(int *d, int l, int r, int v){
while(l <= r){
int m = (l+r)>>1;
// printf("%d %d %d\n", l, r, m);
if(v < d[m]){
r = m - 1;
}else{
l = m + 1;
}
}
return r;
}
bs2,bs3效率不够高,不太严谨,修改如下
int funl(vector<int> &data ,int k) {
int l = 0, r = data.size() - 1;
while(l <= r) {
if(data[l] == k) return l;
int m = l + ((r-l) >> 1);
if(k > data[m]) l = m + 1;
else r = m - 1;
}
if(l >= 0 && l < data.size() && data[l] == k) return l;
else return -1;
}
int funr(vector<int> &data ,int k) {
int l = 0, r = data.size() - 1;
while(l <= r) {
if(data[r] == k) return r;
int m = l + ((r-l) >> 1);
if(k < data[m]) r = m - 1;
else l = m + 1;
}
if(r >= 0 && r < data.size() && data[r] == k) return r;
else return -2;
}