美人松的高度（二分查找）

题目描述

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

输入

第一行，两个正整数N和M（1<=N<=106，1<=M<=103）;
第二行，N个正整数，之间用一个空格隔开，表示N棵美人松的高度；
第三行M个正整数k，表示M个询问，每次询问高度为K的美人松有多少棵，1<=k<=100000。

输出

一行M个正整数，之间用一个空格隔开，分别表示对应每次询问高度为K的树的查询统计数量，不存在则输出0。

样例输入

5 2
2 3 3 4 5
3 4

样例输出

2 1

基本思路：先用二分查找找到符合高度为k的下标，然后向双侧遍历。

代码：

#include <stdio.h>

int bs(long long *ary,int p,int q,long long tar)

{

    if(p>q)return -1;

    int mid=p+(q-p)/2;

    if(ary[mid]==tar)return mid;

    else if(tar<ary[mid])return bs(ary,p,mid-1,tar);

    else return bs(ary,mid+1,q,tar);

}             //查找高度为k的下标

int main ()

{

    int n,m;

    scanf("%d%d",&n,&m);

    long long ary[n+1];

    for(int i=1; i<=n; i++)scanf("%lld",&ary[i]);

    for(int i=1; i<=m; i++)

    {

        int cnt=0;

        long long k;

        scanf("%lld",&k);

        int tmp=bs(ary,1,n,k);

        if(tmp==-1)printf("0 ");

        else

        {

            for(int o=tmp; o<=n; o++)

            {

                if(ary[o]==k)cnt++;

                else break;

            } // 向一侧继续找符合高度为k的数量

            for(int o=tmp; o>=1; o--)

            {

                if(ary[o]==k)cnt++;

                else break;

            }  // 向一侧继续找符合高度为k的数量

            printf("%d ",cnt-1);

        }

    }

return 0;

}