加载中...

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

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

题目描述:

统计一个数字在排序数组中出现的次数。例如,输入排序数组{1,2,3,3,3,3,4,5}和数字3,由于数字3在该数组中出现了4次,所以函数返回4。

解题思路:

既然输入的数组是有序的,所以我们就能很自然的想到用二分查找算法。以题目中给的数组为例,一个比较自然的想法是用二分查找先找到一个3,由于要计算的是输出的次数,所以需要在找到的这个3的左右两边分别再进行顺序扫描,进而得到3的个数,这样最坏的情况下时间复杂度仍然是O(n),和直接顺序扫描的效率相同。

因此,需要考虑怎样更好的利用二分查找算法,由于数组有序,如果知道了第一个k出现的位置和最后一个k出现的位置,那么我们就可以直接算出有多少个k。因此将思路转化为通过二分查找求第一个和最后一个k出现的位置。

以第一个k出现的位置为例,利用二分查找算法可以直接对数组进行二分,而每次总是拿中间的数字和k做比较,如果中间的数字大于k,那么第一个k只有可能出现在左边,下一次直接在数组左半段继续进行二分查找;如果中间的数字小于k,则第一个k只有可能出现在右边,则在右半段再查找;如果中间的数字等于k,我们先判断它前面的一个数字是不是k,如果不是,那么这个中间的数字就是第一个出现的位置,反之,如果中间数字前面的数字是k,那么第一个k仍然在前半段,继续查找。

同理,找最后一个k出现的位置方法类似,可以使用两个函数分别获得。

编程实现(Java):

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
        int first=getFirstIndex(array,k);
        int last=getlastIndex(array,k);
        if(first==-1 || last==-1)
            return 0;
        return last-first+1;
    }
    //二分查找
    public int getFirstIndex(int[] array,int k){
        int res=-1;
        if(array==null||array.length==0)
            return res;
        int low=0,high=array.length-1;
        while(low<=high){ //要有=
            int mid=low+(high-low)/2;
            if(array[mid]<k)
                low=mid+1;
            else if(array[mid]>k)
                high=mid-1;
            else { //相等
                mid=mid-1;
                if(mid<low || array[mid]!=k)
                    return mid+1;
                else 
                    high=mid;
            }
        }
        return res;
    }
    
    public int getlastIndex(int[] array,int k){
        int res=-1;
        if(array==null||array.length==0)
            return res;
        int low=0,high=array.length-1;
        while(low<=high){ //要有=
            int mid=low+(high-low)/2;
            if(array[mid]<k)
                low=mid+1;
            else if(array[mid]>k)
                high=mid-1;
            else { //相等
                mid=mid+1;
                if(mid>high || array[mid]!=k)
                    return mid-1;
                else 
                    low=mid;
            }
        }
        return res;
    }
}
posted @ 2023-06-22 23:07  bujidao1128  阅读(28)  评论(0编辑  收藏  举报