代码源oj 59. 二分查找

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;
}