数字在排序数组中出现的次数

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

看到题目的第一感觉我们肯定想到的是遍历整个数组来求解,但这样的算法时间复杂度是O(N),还不是最优解,我们可以想到使用二分查找来优化。

由于数组是排好序的,假设我们是统计数字K在排序数组中出现的次数,所以我们可以用二分查找直接找到数组中的第一个K和最后一个K。那么如何用二分查找算法在数组中找到第一个K呢?二分查找算法总是先拿中间的数字和K作比较。如果中间的数字比K大,那么K只有可能出现在数组的前半段,下一轮我们只在数组的前半段查找就可以了,如果中间的数字比K小,那么K只有可能出现在数组的后半段,下一轮我们只在数组的后半段查找就可以了。如果中间的数字和K相等呢?我们先判断这个数字是不是第一个K。如果位于中间数字的前面一个数字不是K,此时中间的数字刚好就是第一个K.如果中间数字的前面一个数字也是K,也就是说第一个K肯定在数组的前半段,下一轮我们仍然需要在数组的前半段查找。找到数组中的最后一个K的思路与上述类似。

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

int GetNumOfK(int* data, int length, int k)
{
    int num = 0;
    if (data != NULL && 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)
            num = last - first + 1;
    }
    return num;
}


int main()
{
    int data[] = { 1, 2, 3, 3, 3, 3, 4, 5 };
    int len = sizeof(data) / sizeof(data[0]);
    int ret = GetNumOfK(data, len, 3);
    cout << ret << endl;
    system("pause");
    return 0;
}

这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值