地址 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; }
我的视频题解空间