数字在排序数组中出现的次数 ----《剑指offer》面试题38

题目

统计一个数字在排序数组中出现的次数。

思路

(1)利用二分查找法，查找元素在数组首次出现的位置。
(2)利用二分查找法，查找元素在数组最后出现的位置。
(3)两者相减，就是元素出现的个数。
(4)怎样查找首次出现的位置？二分查找总是拿数组中间的数字和K比较。如果中间的数字和K相等，我们先判断这个数量是不是第一个K。如果位于中间数字的前面一个数字不是K，此时中间的数字刚好就是第一个K。同理，可以查找K最后一个出现的位置。

代码

#include <iostream>

using namespace std;

int GetFirstK(int* data, int length, int k, int start, int end)
{
    if (start > end)
        return -1;

    int middleIndex = (start + end) >> 1;
    int middleData = data[middleIndex];

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

    return GetFirstK(data, length, k, start, end);
}

int GetLastK(int* data, int length, int k, int start, int end)
{
    if (start > end)
        return -1;

    int middleIndex = (start + end) >> 1;
    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 GetLastK(data, length, k, start, end);
}

int GetNumberOfK(int* data, int length, int k)
{
    int number = 0;

    if (data != nullptr && length > 0)
    {
        int first = GetFirstK(data, length, k, 0, length - 1);
        int last = GetLastK(data, length, k, 0, length - 1);

        if (first > -1 && last > -1)
            number = last - first + 1;
    }

    return number;
}

int main()
{
    int arr[] = {1,2,3,3,3,3,4,5};
    int k = 3;
    cout << "The occurrence of K is: "
        << GetNumberOfK(arr, k, sizeof(arr) / sizeof(int));

    return 0;
}