ACM二分算法模板 lower_bound upper_bound 具体实现（四个模板）

ACM二分算法模板 （四个模板）

例如给出一个数组（或者是一种对应关系），查询时x会有四种返回结果

一 . 返回大于等于x的位置（即lower_bound）

划分区间为【l，mid】,【mid +1 , r】区间向着答案的右边靠近 ， 如果说当答案刚好落在mid的位置上，如果此时判断落在【mid +1 , r】就是大于答案的下标，如果说落在【l , mid】 上就是返回大于等于答案的坐标

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+7;
typedef long long ll;
int A[10] = {1,3,4,5,7,9,10,12,16,21}; 
int main(){
	int l = 0,r = 10;
	int want ;cin>>want;
	while(r>l){
		int m = l+r>>1;
		if(A[m]>=want) r = m; //刚好等于时落在左边区间就是大于等于
		else l = m+1;
	}
	cout<<A[l];
	return 0;
}

二.返回大于x的位置（即upper_bound）

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+7;
typedef long long ll;
int A[10] = {1,3,4,5,7,9,10,12,16,21}; 
int main(){
	int l = 0,r = 10;
	int want ;cin>>want;
	while(r>l){
		int m = l+r>>1;
		if(A[m]>want) r = m;刚好等于时落在右边区间就是大于等于
		else l = m+1;
	}
	cout<<A[l];
	return 0;
}

三.返回小于x的位置

如果说返回的是答案的左边，那么划分区间应该是【l,mid-1】,【mid, r】
如果说等于mid落在左边区间，那么返回的是小于x的下标，如果等于mid
落在右边区间，那么返回的是小于等于x的下标

这里还要注意 mid = l+r+1>>1 ,这里加上1 是要防止二分进入死循环
否则在l=0，r = 1时会卡死

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+7;
typedef long long ll;
int A[10] = {1,3,4,5,7,9,10,12,16,21}; 
int main(){
	int l = 0,r = 10;
	int want ;cin>>want;
	while(r>l){
		int m = l+r+1>>1;
		if(A[m]>=want) r = m-1;// 等于落在左边区间就是小于查找值的位置
		else l = m;
	}
	cout<<A[l];
	return 0;
}

四.返回小于等于x的位置

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+7;
typedef long long ll;
int A[10] = {1,3,4,5,7,9,10,12,16,21}; 
int main(){
	int l = 0,r = 10;
	int want ;cin>>want;
	while(r>l){
		int m = l+r+1>>1;
		if(A[m]>want) r = m-1;// 等于落在右边区间就是小于查找值的位置
		else l = m;
	}
	cout<<A[l];
	return 0;
}