剑指offer面试题38 数在有序数组中出现的次数，二分找第一个和最后一个出现的这个数字

/*
这题太棒了。。剑指offer是我大学生涯读过最好的一本书了。
看到有序数组 找一个数肯定是想到二分 找到后 前后探查 看一共多少个。
但这样前后探查 复杂度和遍历一遍一样 都是O(n) 肯定是不好的
我们希望这个复杂度是logn 这样的话。。想到找到数组中第一个k和最后一个k
那就是改一下二分的条件
写两个二分 分别找第一个k和最后一个k
找第一个的时候 如果找到了k 看是不是第一个 是就返回下标
不是的话 第一个还在左边
右边类似
找到k看是不是最后一个 不是的话 肯定在右边
题目很不错啊


*/
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=1000;

int GetFirstK(int * arr,int len,int k)
{
    int start=0,end=len-1,mid;
    while(start<=end)
    {
        mid=(start+end)>>1;
        if(arr[mid]>k)
        end=mid-1;
        else if(arr[mid]<k)
        start=mid+1;
        else
        {
            if(mid==0 || (arr[mid-1]!=k))
            return mid;
            else
            end=mid-1;
        }
    }
    return -1;
}

int GetLastK(int * arr,int len,int k)
{
    int start=0,end=len-1,mid;
    while(start<=end)
    {
        mid=(start+end)>>1;
        if(arr[mid]>k)
        end=mid-1;
        else if(arr[mid]<k)
        start=mid+1;
        else{
            if(mid==len-1 || arr[mid+1]!=k)
            return mid;
            else
            start=mid+1;
        }
    }
    return -1;
}

int GetNumberOfK(int * arr,int len,int k)
{
    if(arr==NULL || len<=0)
    return 0;
    int firstindex=GetFirstK(arr,len,k);
    int lastindex=GetLastK(arr,len,k);
    //cout<<firstindex<<" "<<lastindex<<endl;
    if(firstindex==lastindex && firstindex==-1)
    return 0;
    return lastindex-firstindex+1;
}

int main()
{
    freopen("/home/gl/in","r",stdin);
    int n,k,arr[100];
    while(scanf("%d %d",&n,&k)!=EOF)
    {
        for(int i=0;i<n;++i)
        scanf("%d",&arr[i]);
        printf("%d\n",GetNumberOfK(arr,n,k));
    }
    return 0;
}