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

原创于 2020-07-27 16:25:38 发布 · 160 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一种高效算法，用于统计排序数组中特定数字的出现频率。通过使用二分查找，该算法能够快速定位目标数字，并通过扩展的二分查找确定其范围，从而计算出出现次数。

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

排序数组，所以相比顺序，用二分更高效

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
       int len=array.length;
       int l=0,r=len-1;
       int num=0;
       while(l<=r){
           int mid=l+(r-l)/2;
           if(k>array[mid]){
               l=mid+1;
           }else if(k<array[mid]){
               r=mid-1;
           }else{
               num=howMuch(array,mid,k);
               break;
           }
       }
        return num;
    }
    public int howMuch(int [] array,int mid,int k){
        int num=1,l=mid-1,r=mid+1;
        while(l>=0&&array[l]==k){
            num++;
            l--;
        }
        while(r<array.length&&array[r]==k){
            num++;
            r++;
        }
        return num;
    }
}

更新颖高效的解法，在二分基础之上的扩展
因为data中都是整数，所以可以稍微变一下，不是搜索k的两个位置，而是搜索k-0.5和k+0.5
这两个数应该插入的位置，然后相减即可。

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
        int len=array.length;
        int l=findIndex(len,array,(double)k-0.5);
        int r=findIndex(len,array,(double)k+0.5);
        return r-l;
    }
    public int findIndex(int len,int[] array,double target){
        int l=0,r=len-1;
        while(l<=r){
            int mid=l+(r-l)/2;
            if((double)array[mid]<target){
                l=mid+1;
            }else if((double)array[mid]>target){
                r=mid-1;
            }
        }
        return l;
    }
}