中石油OJ: 美人松高度2

【问题描述】
又到过年了，狗熊岭的动物们都忙碌起来，张灯结彩准备过年。李老板却要光头强砍掉一些百年美人松回去。美人松都是很高的，但是也不会超过长整型（long long)。现在光头强看到丛林里有N颗美人松按照从矮到高排好了，当然每棵松的高度都是已知的。李老板要问光头强M次：每次询问高度为K的美人松是否存在？

【输入】
第一行，两个正整数N，M,1<=N<=10^ 6,1<=M<=10^3;
第二行，N个正整数，之间用一个空格隔开，表示N棵美人松的高度；
第三行M个正整数k，表示M个询问，每次询问高度为K的美人松是否存在。
（提示：要用scanf读入，否则会超时）

【输出】
一行M个正整数，之间用一个空格隔开，分别表示对应每次询问高度为K的树的查询结果，如果存在，则输出1，不存在则输出0

【样例输入】

5 2
2 3 3 4 5
3 6

【样例输出】

1 0

【思路及题目分析】
数据量大小是1e6，并且有1e3次的询问操作，因此每次操作的时间复杂度必须控制在logn级别，因此用整数二分。
个人选用板子为：找到大于x的最小的数

int bin1(int l, int r, int x) {          //找到大于x的最小的数
	while (l < r) {
		int mid = (l + r) >> 1;
		if (q[mid] > x) r = mid;        //注：使用方法需要判定返回的下标是否在合理范围内
		else l = mid + 1;
	}
	return l;
}

AC代码：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

map<int, int> m;    //<小组大小，相同数量的人群>
int n;
int res = 0;

int main() {
    scanf("%d", &n);
    for(int i = 0; i < n; i ++) {
        int x;
        scanf("%d", &x);
        x ++;       //把自己算在内
        m[x] ++;    //说相同数量形式的学生增加了一个
    }
    
    for(auto x : m) {
        int t = x.first;
        int cnt = x.second;
        
        int group = cnt / t;
        int remain = (cnt % t == 0) ? 0 : t;
        
        int total = (group * t) + remain;
        res += total;
    }
    
    cout << res << endl;
    
    return 0;
}