剑指offer-37：数字在排序数组中出现的次数

最新推荐文章于 2024-07-21 08:44:26 发布

原创最新推荐文章于 2024-07-21 08:44:26 发布 · 133 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

67 篇文章

订阅专栏

本文介绍了一种高效算法，用于统计一个数字在已排序数组中出现的次数。通过使用二分查找法，该算法能够快速定位首个和末位目标数字，从而计算其在数组中的总出现次数。

题目描述

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

思路

看见有序，肯定就是二分查找了，算法比较简单，不多说，值得一提的是，不要拘泥于递归，要会循环写法。

代码

public class Solution37 {


    public int GetNumberOfK(int[] array, int k) {
        int length = array.length;
        if (length == 0) {
            return 0;
        }
        int firstK = getFirstK(array, k, 0, length - 1);
        int lastK = getLastK(array, k, 0, length - 1);
        if (firstK != -1 && lastK != -1) {
            return lastK - firstK + 1;
        }
        return 0;
    }

    //递归写法
    private int getFirstK(int[] array, int k, int start, int end) {
        if (start > end) {
            return -1;
        }
        int mid = (start + end) >> 1;
        if (array[mid] > k) {
            return getFirstK(array, k, start, mid - 1);
        } else if (array[mid] < k) {
            return getFirstK(array, k, mid + 1, end);
        } else if (mid - 1 >= 0 && array[mid - 1] == k) {
            return getFirstK(array, k, start, mid - 1);
        } else {
            return mid;
        }
    }

    //循环写法
    private int getLastK(int[] array, int k, int start, int end) {
        int length = array.length;
        int mid = (start + end) >> 1;
        while (start <= end) {
            if (array[mid] > k) {
                end = mid - 1;
            } else if (array[mid] < k) {
                start = mid + 1;
            } else if (mid + 1 < length && array[mid + 1] == k) {
                start = mid + 1;
            } else {
                return mid;
            }
            mid = (start + end) >> 1;
        }
        return -1;
    }


    public static void main(String[] args) {

        BeanUtil.print(new Solution37().GetNumberOfK(new int[]{1, 2, 3, 3, 3, 3, 5}, 3));

    }

}