二分查找，实现自定义lower_bound函数、upper_bound函数

最新推荐文章于 2024-04-26 22:11:43 发布

原创最新推荐文章于 2024-04-26 22:11:43 发布 · 1.3k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

本文介绍了二分查找算法中的两种核心实现：lower_bound和upper_bound函数。通过对这些函数的讲解及代码实现，帮助读者理解如何在有序数组中找到特定元素的第一个出现位置或最后一个出现位置之后的位置。适用于算法初学者及面试准备。

lower_bound函数

#include<iostream>
#include<algorithm>
using namespace std;
const int M=1e5;

int my_lower_bound(int *a,int n,int x){
	int l=0,r=n-1;
	int m;
	while(l<r){
		m=(l+r)/2;
		if(a[m]<x) l=m+1;
		else r=m;
	}
	if(a[l]>=x) return l;
	else return -1;
}

int main()
{
	int a[M],n,x;
	cin >> n >> x;
	for(int i=0;i<n;i++)
		cin >> a[i] ;
	sort(a,a+n);
	cout << my_lower_bound(a,n,x);
	return 0;
}

upper_bound函数

#include<iostream>
#include<algorithm>
using namespace std;
const int M=1e5;

int my_upper_bound(int *a,int n,int x){
	int l=0,r=n-1;
	int m;
	while(l<r){
		m=(l+r)/2;
		if(a[m]<=x) l=m+1;
		else r=m;
	}
	if(a[l]>x) return l;
	else return n;
}

int main()
{
	int a[M],n,x;
	cin >> n >> x;
	for(int i=0;i<n;i++)
		cin >> a[i] ;
	sort(a,a+n);
	cout << my_upper_bound(a,n,x);
	return 0;
}