地址 https://www.papamelon.com/problem/257
二分查找模板题
二分有两个模板
一个是找到符合查询数字的下边界<=
一个是找到符合查询数字的上边界>=
记熟会方便很多.
本题是查找第一个大于等于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;
}