papamelon 257. 下界 lower_bound(挑战程序设计竞赛)

地址 https://www.papamelon.com/problem/257
papamelon 257. 下界 lower_bound(挑战程序设计竞赛)

二分查找模板题
二分有两个模板
一个是找到符合查询数字的下边界<=
一个是找到符合查询数字的上边界>=
记熟会方便很多.

本题是查找第一个大于等于k的数字
是下边界

#include <iostream>

using namespace std;

const int N = 1000010;
int arr[N];

int n, k;

int main()
{
	cin >> n >> k;
	for (int i = 0; i < n; i++) { cin >> arr[i]; }
	int l = 0; int r = n ;
	while (l < r) {
		int mid = (l + r) >> 1;
		if (arr[mid] >= k) { r = mid; }
		else { l = mid + 1; }
	}
	cout << l << endl;

	return 0;
}

也可以使用stl的二分函数api
注意 lower_bound和upper_bound 分别等于数组中大于等于查找元素的位置 和数组中第一个大于查找元素的位置.

#include <iostream>

using namespace std;

const int N = 1000010;
int arr[N];

int n, k;

int main()
{
	cin >> n >> k;
	for (int i = 0; i < n; i++) { cin >> arr[i]; }
	int ans = lower_bound(arr,arr+n,k) -arr;
	cout << ans<<endl;

	return 0;
}

我的视频题解空间

上一篇:归并排序非递归版本2及其相关面试题


下一篇:lower_bound 小知识