【剑指Offer-Java】数字在排序数组中出现的次数

题目描述

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

思路

因为是排序数组，所以想到用二分查找法，只要找到第一个和最后一个k，那这个范围内都是k了

本题目的是想直接利用二分查找法找到第一个k和最后一个k
找第一个k步骤: 拿数组中间数字m和k比较，
（1）若m>k，k只可能出现在数组前半段，下一轮只需在前半段找
（2）若m<k，k只可能出现在数组前后段，下一轮只需在后半段找
（3）若m=k，先判断这个中间数字m是不是第一个k，
若中间前面一个数字不是k，则m就是第一个k；
若中间数字前一个是k，则第一个k肯定再数组前半段，下一轮仍然需要再在数组前半段找

找最后一个k的思路类似

实现

//二分查找法，找到第一个k和最后一个k
    public int GetNumberOfK(int [] array , int k) {
        int FirstK=getFirstK(array,k,0,array.length-1);
        int LastK=getLastK(array,k,0,array.length-1);
        if(FirstK!=-1 && LastK!=-1){
            return LastK-FirstK+1;
        }
        return 0;
    }
    //获取第一个k:循环写法
    public int getFirstK(int[] arr,int k,int start,int end){
        int mid=(end-start)/2+start;
        while(start<=end){
            if(arr[mid]>k){
                end=mid-1;
            }
            else if(arr[mid]<k){
                start=mid+1;
            }
            else if(mid-1>=0 && arr[mid-1]==k){
                end=mid-1;
            }
            else{
                return mid;
            }
            mid=(end-start)/2+start;
        }
        return -1;
    }
    
    //获取第一个k:递归写法
    public int getLastK(int[] arr,int k,int start,int end){
        //递归出口
        if(start>end) return -1;
        int mid=(end-start)/2+start;
        if(arr[mid]>k){
            return getLastK(arr,k,start,mid-1);
        }
        else if(arr[mid]<k){
            return getLastK(arr,k,mid+1,end);
        }
        else if(mid+1<arr.length && arr[mid+1]==k){
            return getLastK(arr,k,mid+1,end);
        }
        else{
            return mid;
        }
    }