二分(整数和浮点数)
1.整数:
核心代码:
bool check(int x) {/* ... */} // 检查x是否满足某种性质
// 区间[l, r]被划分成[l, mid]和[mid + 1, r]时使用:
int bsearch_1(int l, int r){
while (l < r){
int mid = l + r >> 1;
if (check(mid)) r = mid; // check()判断mid是否满足性质
else l = mid + 1;
}
return l;
}
// 区间[l, r]被划分成[l, mid - 1]和[mid, r]时使用:
int bsearch_2(int l, int r){
while (l < r){
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
return l;
}
题目789:数的范围
给定一个按照升序排列的长度为n的整数数组,以及 q 个查询。
对于每个查询,返回一个元素k的起始位置和终止位置(位置从0开始计数)。如果数组中不存在该元素,则返回“-1 -1”。
输入格式
第一行包含整数n和q,表示数组长度和询问个数。
第二行包含n个整数(均在1~10000范围内),表示完整数组。
接下来q行,每行包含一个整数k,表示一个询问元素。
输出格式
共q行,每行包含两个整数,表示所求元素的起始位置和终止位置。
如果数组中不存在该元素,则返回“-1 -1”。
数据范围
1≤n≤100000
1≤q≤10000
1≤k≤10000
输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1
//version 1.0
#include<iostream>
using namespace std;
const int N = 100010;
int n, m, k, q[N]; //m是询问次数, k为询问元素
int main(){
cin >> n >> m;
for(int i = 0; i < n; ++i) cin >> q[i];
while(m--){ //m次询问
cin >> k;
int l = 0, r = n - 1;
//循环结束的结果是找到k出现的起始位置l,且l = r,若该元素不存在则找到q中第一个大于k的数。
while(l < r){
int mid = (l + r) / 2;
if(q[mid] >= k) r = mid;
else l = mid + 1;
}
//判断所找到的数是否为k,是则输出该数所在位置,否则输出-1 -1
if(q[l] != k) cout << "-1 -1" << endl;
else{
cout << q[l] << " ";
l = 0, r = n - 1; //补1,避免死循环
//循环结束的结果是找到k出现的最终位置l,且l = r,若该元素不存在则找到q中第一个大于k的数。
while(l < r){
int mid = (l + r + 1) / 2;
if(q[mid] <= k) l = mid;
else r = mid - 1;
}
cout << l << endl;
}
}
return 0;
}
//version 2.0
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int n, m, k, q[N];
int bsearch_l(int l, int r){
while(l < r){
int mid = l + r >> 1;
if(q[mid] >= k) r = mid;
else l = mid + 1;
}
if(q[l] != k) return -1;
else return l;
}
int bsearch_r(int l, int r){
while(l < r){
int mid = l + r + 1>> 1;
if(q[mid] <= k) l = mid;
else r = mid - 1;
}
if(q[r] != k) return -1;
else return r;
}
int main(){
cin >> n >> m;
for(int i = 0; i < n; ++i) cin >> q[i];
while(m--){
cin >> k;
int l = 0, r = n - 1;
cout << bsearch_l(l, r) << " " << bsearch_r(l, r) << endl; //注意输出格式
}
return 0;
}
2.浮点数:
核心代码:
bool check(double x) {/* ... */} // 检查x是否满足某种性质
double bsearch_d(double l, double r){
const double eps = 1e-8; // eps 表示精度,取决于题目对精度的要求,一般要求保留6位小数,则精度为1e-8;若要求保留5为小数,则精度为1e-7
while (r - l > eps){
double mid = (l + r) / 2;
if (check(mid)) r = mid;
else l = mid;
}
return l;
}
题目790:数的三次方根
给定一个浮点数n,求它的三次方根。
输入格式
共一行,包含一个浮点数n。
输出格式
共一行,包含一个浮点数,表示问题的解。
注意,结果保留6位小数。
数据范围
-10000≤n≤10000
输入样例:
1000.00
输出样例:
10.000000
#include<stdio.h>
double n;
double bsearch_d(double l, double r){ //核心代码
const double eps = 1e-8; //设置精度
while(r - l > eps){
double mid = (l + r) / 2;
if(mid * mid * mid >= n) r = mid;
else l = mid;
}
return l;
}
int main(){
scanf("%lf", &n);
//注意:由于当n<1时,其三次方根大于n,故搜索范围不能取[-n, n],此处即用所给数据范围[-1000, 1000] .
printf("%lf", bsearch_d(-1000, 1000));
return 0;
}