501 二分查找

// 501 二分查找.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

/*
http://oj.daimayuan.top/course/22/problem/59


有 n个数 a1,a2,…,an和 m个询问。

每次询问给一个数 x,回答 a1,a2,…,an
里面分别有多少个数字小于 x,等于 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
*/


#include <iostream>
#include <algorithm>

using namespace std;

int n, m;
const int N = 200010;
int arr[N];

void solve() {
	int x; 
	//cin >> x;
	scanf("%d",&x);

	int ansl = 0; int ansr = 0; int ansm = 0;
	int a = 0, b = n+1;
	while (a < b) {
		int mid = a + b >> 1;
		if (arr[mid] >= x) b = mid; 
		else a = mid + 1;
	} 
	ansl = a - 1;
	
	int l = 0; int r = n+1;
	while (l < r) {
		int mid = (l + r + 1) >> 1;
		if (arr[mid] <= x) l = mid; else r = mid - 1;
	}  
	ansr = n - l;
	
	//cout << ansl<<" " << n - ansl - ansr << " " << ansr << endl;
	printf("%d %d %d\n", ansl, n - ansl - ansr,ansr);
}

int main()
{
	//cin >> n >> m;
	scanf("%d%d",&n,&m);
	//for (int i = 1; i <= n; i++) cin >> arr[i];
	for (int i = 1; i <= n; i++) scanf("%d", &arr[i]);
	sort(&arr[1], &arr[1]+n);
	arr[n + 1] = 0x3f3f3f3f;
	for (int i = 1; i <= m; i++) {
		solve();
	}

	return 0;
}

 

posted on 2024-11-29 17:49  itdef  阅读(11)  评论(0)    收藏  举报

导航