欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2024-03-20 13:00:22
...

剑指offer中的题目


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


看到有序数组,可能很多人都直接想到了要使用二分查找的方法,实际上确实是使用二分查找,但是需要稍微的修改一下,中间索引对应的元素等于目标元素的情况。(这里将二分数组中间索引对应的元素值,简称为中间值)
一般二分法,如果中间值大于目标值,则目标值在数组的右半部分;如果中间值小于目标值,则目标值在数组的左半部分;如果中间值等于目标值,则直接返回中间值的索引
这也是这道题最大的陷阱,可能会有如下数组样本,

[1,2,3,3,3,3,4,5,7]

所以在遇到中间等于目标值的情况的时候,需要对数组的左右两边的部分都进行二分递归查找,而不是返回或者只查找数组左边和右边部分中的一个。

下面是代码

public class solution {

    public static int GetNumberOfK(int[] array , int k) {
        if(array.length == 0 || array == null){
            return 0;
        }else if(array.length == 1 && array[0] == k){
            return 1;
        }
        return GetNumberOfKSupport(array, 0, array.length-1, k);
    }

    private static int GetNumberOfKSupport(int[] array, int start, int end, int k){
        if(start == end){
            return 0;
        }/*else if(start == end && array[start] == k){
            return 1;
        }*/

        int mid = start + (end - start) / 2;
        int leftCount = 0;
        int rightCount = 0;
        if(array[mid] == k){
            leftCount = GetNumberOfKSupport(array, start, mid, k);
            rightCount = GetNumberOfKSupport(array, mid + 1, end, k);
        }else if(array[mid] > k){  //在数组的左边
            leftCount = GetNumberOfKSupport(array, start, mid, k);
        }else{ // array[mid] < k //在数组的右边
            rightCount = GetNumberOfKSupport(array, mid + 1, end , k);
        }

        int count = 0;
        for(int i = start; i <= end; i++){
            if(array[i] == k){
                count++;
            }
            array[i] = k+1;
        }

        return count + leftCount + rightCount;
    }

    public static void main(String[] args) {
        int[] array = {3,3,3,3,3,3};
        int k = 3;
        int number = GetNumberOfK(array, k);
        System.out.println(number);
    }
}

结果显示

6