59. 二分查找
有n个数a1,a2,…,an和m个询问。
每次询问给一个数x,回答这些数字里面有多少个数字小于,等于,大于x。
输入格式
第一行两个整数n,m。接下来一行n个整数,表示a1,a2,…,an。
接下来m行,每行一个整数x。
输出格式
输出m行,每行三个整数。
样例输入
6 6
3 5 1 5 3 5
1
2
3
4
5
6
样例输出
0 1 5
1 0 5
1 2 3
3 0 3
3 3 0
6 0 0
数据规模
对于100%的数据,满足1≤n,m≤2×105,1≤ai,x≤109。
思路
二分法,先排序,之后在每次询问下二分,(l, r) 的开区间,所以输入时数组下标从1 开始
只要 (l + 1) < r 就继续循环,最后 l 指向的下标就是我们想要的
分成三种情况
1.小于x
2.等于x
3.大于x
1.小于x:取中间的下标取值,如果a[m] < x 说明 l 太小啦,所以 l = m,否则的话就让 r = m,最后得到的 l 的下标就是小于 x 的数的个数
2.等于x,这个可以不算,等一下因为可能整个数组都 = x 所以得遍历一遍,我们先算 大于x的个数,然后用数组长度 n - 小于x - 大于x 就可以得到等于x 的数的个数了
3.大于x:这个和第一种情况类似,只不过把 大于x 看成一种情况,小于等于 x 合并为另一种情况来对 l 或者 r 赋值,类比第一种情况,这里最后 l 的下标指向的数就是最后一个小于等于x 的数,则n - l 就是大于x的数的个数
代码
#define _CRT_SECURE_NO_WARNINGS
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
int a[200001];
int main() {
int n, m, x;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++)
scanf("%d", &a[i]);
sort(a + 1, a + 1 + n);
for (int i = 1; i <= m; i++) {
scanf("%d", &x);
int maxn = 0, minn = 0, equ = 0;
int l = 0, r = n + 1;
while (l + 1 < r) {
int m = (l + r) >> 1;
if (a[m] < x)
l = m;
else
r = m;
}
minn = l;
l = 0, r = n + 1;
while (l + 1 < r) {
int m = (l + r) >> 1;
if (a[m] > x)
r = m;
else
l = m;
}
maxn = n - l;
equ = n - maxn - minn;
printf("%d %d %d\n", minn, equ, maxn);
}
return 0;
}