剑指offer面试题目：数字在排序数组中出现的次数

题目

统计一个数字在排序数组中出现的次数。例如，输入排序数组{1,2,3,3,3,3,4,5}和数字3由于3在这个数组中出现了4次，因此输出4。

---

分析

题目解法很多，关键是要找到让人满意的方法，直接统计当然可以，可是显然不是我们要的答案。
比较好的思路如下：
使用二分查找的拓展，当查找的元素有重复的时，找到元素的第一个和最后一个。这样将可以计算出该元素有多少个重复的了。二分法在数组中查找一个合乎要求的数字时间复杂度是O(logn)，因此总的时间复杂度也只有O(logn)。

---

代码

int GetFirstK(int *data, int length, int k, int start, int end)  
{  
    if(start>end)  
        return -1;  
    int middleIndex=(start+end)/2;  
    int middleData=data[middleIndex];  

    if (middleData==k)  
    {  
        //判断是否是第一个k  
        if ((middleIndex>0&&data[middleIndex-1]!=k)||middleIndex==0)  
        {  
            return middleIndex;  
        }  
        else  
        {//第一个k肯定在左边  
            end=middleIndex-1;  
        }  
    }  
    else if (middleData>k)  
    {  
        end=middleIndex-1;  
    }  
    else  
        start=middleIndex+1;  
    //递归  
    return GetFirstK(data,length,k,start,end);  
}  

//同样的思路，递归找到最后的一个k  
int GetEndK(int *data, int length, int k, int start, int end)  
{  
    if(start>end)  
        return -1;  
    int middleIndex=(start+end)/2;  
    int middleData=data[middleIndex];  

    if (middleData==k)  
    {  
        if ((middleIndex<length-1&&data[middleIndex+1]!=k)||middleIndex=length-1)  
        {  
            return middleIndex;  
        }  
        else  
            start=middleIndex+1;  
    }  
    else if (middleData>k)  
    {  
        end=middleIndex-1;  
    }  
    else  
    {  
        start=middleIndex+1;  
    }  
    return GetEndK(data,length,k,start,end);  
}  

//计算出k在数组中出现的次数  
int GetNumOfK(int *data, int length, int k)  
{  
    int number=0;  
    if (data!=NULL&&length>0)  
    {  
        int first=GetFirstK(data,length,k,0,length-1);  
        int last=GetEndK(data,length,k,0,length-1);  
        if (first>-1&&last>-1)  
        {  
            return last-first+1;  
        }  
    }  
    return number;  
}  

上面的采用了递归，下面采用循环效果更好，但解题思路一样。

#include<stdio.h>
#include<stdlib.h> 
//全局变量 
int arr[] = {0,1,1,2,2,2,2,4,4,4}; //测试数组 
int size = sizeof(arr) / sizeof(*arr); //数组元素个数 
int getUpper(int arr[], int key){//获取某个元素最后出现位置 
    int low = 0, high = size - 1;
    //其实是一个递归迭代 
    while(low < high){
        int mid = (low + high + 1) / 2;
        if(arr[mid] <= key)
        //当要查找的值比中位数大于等于时，把查找的低位限制为mid 
            low = mid;
        else
        //当要找的值比 中位数小时，，把查找的高位限制为mid-1 
            high = mid - 1;          
    } 
//    返回最后出现位置 
    return low; 
}
int getLower(int arr[], int key){//获取某个元素第一次出现位置 
    int low = 0, high = size - 1;
    while(low < high){
        int mid = (low + high) / 2;
        //当要找的值比 中位数小于等于时，，把查找的高位限制为mid+1 
        if(arr[mid] >= key)
            high = mid;
        else
        //当要找的值比 中位数大时，，把查找的低位限制为mid+1 
            low = mid + 1;           
    }    
    //返回第一次出现位置 
    return low; 
} 
int main(){
    for(int i = 0; i < size; i ++)
        printf("%d 出现 %d 次\n",arr[i],getUpper(arr,arr[i]) - getLower(arr,arr[i]) + 1);
    getchar();  
}