几种不同的二分查找
程序员文章站
2022-03-14 23:19:10
...
今天多校又卡在找下标上了,队友贡献了一发自己写的二分板子,感觉很好用。保存下来。
/*
几种二分方法整理
元素可以重复
*/
//lower_bound(num, num+size, x)-num:大于等于x的第一个数的下标
//upper_bound(num, num+size, x)-num:大于x的第一个数的下标
//1.求等于x的最小的index,不存在返回-1
int binary (int *num, int start, int end, int x) {
int l = start, r = end, ans=-1;
while(l <= r) {
int mid = (l+r) >> 1;
if(num[mid] == x) {
ans = mid;
r = mid - 1;
}
else if(num[mid] > x)
r = mid - 1;
else
l = mid + 1;
}
return ans;
}
//2.求等于x的最大的index,不存在返回-1
int binary (int *num, int start, int end, int x) {
int l = start, r = end, ans=-1;
while(l <= r) {
int mid = (l+r) >> 1;
if(num[mid] == x) {
ans = mid;
l = mid + 1;
}
else if(num[mid] > x)
r = mid - 1;
else
l = mid + 1;
}
return ans;
}
//3.求小于x的最大的index
int binary (int *num, int start, int end, int x) {
int l = start, r = end;
while(l <= r) {
int mid = (l+r) >> 1;
if(num[mid] >= x)
r = mid - 1;
else
l = mid + 1;
}
return r;
}
//4.求大于x的最小的index
int binary (int *num, int start, int end, int x) {
int l = start, r = end;
while(l <= r) {
int mid = (l+r) >> 1;
if(num[mid] <= x)
l = mid + 1;
else
r = mid - 1;
}
return l;
}
//5.求大于等于x的最小的index
int binary (int *num, int start, int end, int x) {
int l = start, r = end;
while(l <= r) {
int mid = (l+r) >> 1;
if(num[mid] >= x)
r = mid - 1;
else
l = mid + 1;
}
return l;
}
//6.求小于等于x的最大的index
int binary (int *num, int start, int end, int x) {
int l = start, r = end;
while(l <= r) {
int mid = (l+r) >> 1;
if(num[mid] <= x)
l = mid + 1;
else
r = mid - 1;
}
return r;
}
下一篇: 二分查找的循环与递归实现