面试题38:数字在排序数组中出现的次数

来源:互联网 发布:编写手机app软件 编辑:程序博客网 时间:2024/06/08 11:21

问题:统计一个数字在排序数组中出现的次数。例如输入排序数组{1,2,3,3,3,3,4,5}和数字3,

由于3在这个数组中出现了4次,因此输出4。


思路1:因为数组是排序的,所以我们可以利用折半查找来找到我们要找的这个数字。

找到这个数字之后,从左右两边依次开始继续检测,找到最左边和最右边的那个数字,相减就可以得到次数。

但是这样做的时间复杂度为O(n),


思路2:利用折半查找找到我们要找的元素之后,继续使用折半查找找到第一个和最后一个我们需要找到的元素。

int getFirstK(int *a, int length, int start, int end, int k){    if(start > end)        return -1;        int mid = (end + start) / 2;    int midData = a[mid];        if(midData == k)    {        if(mid == 0 || a[mid-1] != k)            return mid;        else            end = mid - 1;    }        else if(midData > k)        end = mid - 1;    else        start = mid + 1;    return getFirstK(a, length, start, end, k);}int getLastK(int *a, int length, int start, int end, int k){    if(start > end)        return -1;        int mid = (end + start) / 2;    int midData = a[mid];        if(midData == k)    {        if(mid == length - 1 || a[mid+1] != k)            return mid;        else            start = mid + 1;    }        else if(midData > k)        end = mid - 1;    else        start = mid + 1;        return getLastK(a, length, start, end, k);}int getResult(int *a, int length, int start, int end, int k){    int first = getFirstK(a, length, start, end, k);    int last = getLastK(a, length, start, end, k);        if(first < 0 || last < 0)        return -1;        return last - first + 1;}

0 0
原创粉丝点击