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

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

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

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

public class Offer39 {

    //找出第一个k的位置
    private int getFirstK(int nums[], int k, int start, int end) {
        if (start > end) {
            return -1;
        }
        int middleIndex = (start + end) / 2;
        int middleData = nums[middleIndex];
        if (middleData == k) {
            //找出第一个位置k的终止条件:要么middleIndex=0;要么middleIndex > 0但是middleIndex-1处的位置不等于k
            if ((middleIndex > 0 && nums[middleIndex - 1] != k) || middleIndex == 0) {
                return middleIndex;
            } else {
                end = middleIndex - 1;
            }
        } else if (middleData > k) {
            end = middleIndex - 1;
        } else {
            start = middleIndex + 1;
        }
        return getFirstK(nums, k, start, end);
    }

    //找出最后一个k的位置
    private int getLastK(int nums[], int k, int start, int end) {
        if (start > end) {
            return -1;
        }
        int middleIndex = (start + end) / 2;
        int middleData = nums[middleIndex];
        if (middleData == k) {
            //找出最后一个位置k的终止条件:要么middleIndex=长度-1;要么middleIndex < len-1但是middleIndex+1处的位置不等于k
            if ((middleIndex < nums.length - 1 && nums[middleIndex + 1] != k) || middleIndex == nums.length - 1) {
                return middleIndex;
            } else {
                start = middleIndex + 1;
            }
        } else if (middleData < k) {
            start = middleIndex + 1;
        } else {
            end = middleIndex - 1;
        }
        return getLastK(nums, k, start, end);
    }

    public int getNumberOfK(int nums[], int k) {
        int number = 0;
        int len = nums.length;
        if (nums != null && len > 0) {
            int firstK = getFirstK(nums, k, 0, len - 1);
            int lastK = getLastK(nums, k, 0, len - 1);
            if (firstK > -1 && lastK > -1) {
                number = lastK - firstK + 1;
            }
        }
        return number;
    }

    public static void main(String[] args) {
        Offer39 m1 = new Offer39();
        int nums[] = new int[]{1, 2, 3, 3, 3, 3, 4, 5};
        int count = m1.getNumberOfK(nums, 3);
        System.out.println(count);
    }