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

比天空更远

于 2019-01-08 11:52:50 发布

阅读量193

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer 剑指offer

本文链接：https://blog.youkuaiyun.com/weixin_42449534/article/details/86065543

剑指offer 同时被 2 个专栏收录

66 篇文章

订阅专栏

剑指offer

66 篇文章

订阅专栏

本文介绍了一种在排序数组中查找特定元素出现次数的方法，通过二分查找定位目标元素，并进一步确定其首次和末次出现的位置，以此计算元素出现的总次数。文章提供了详细的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一看到排序数组，首先要想到用二分法来解决

二分查找，然后向前向后找

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
        if(array == null || array.length <= 0)
            return 0;
        int count = 0;
        int low = 0, high = array.length-1, mid = 0;
        while(low <= high){
            mid = low + (high - low) / 2;
            if(array[mid] == k ){
                break;
            }
            else if(array[mid] > k){
                high = mid - 1;
            }
            else 
                low = mid + 1;
        }
        if(low <= high){
            for(int i = mid; i >= 0; i-- ){
                if(array[i] == k)
                    count++;
                else
                    break;
            }
            for(int i = mid + 1; i < array.length; i++){
                if(array[i] == k)
                    count++;
                else
                    break;
            }
        }
        return count;
    }
}

有一种改进是利用二分查找找到该数字在数组中第一次、最后一次出现的位置。copy来的代码：

链接：https://www.nowcoder.com/questionTerminal/70610bf967994b22bb1c26f9ae901fa2
来源：牛客网

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
        if(array == null || array.length == 0)
            return 0;
        int first = getFirstK(array,k,0,array.length - 1);
        int last = getLastK(array,k,0,array.length - 1);
        if(first == -1 || last == -1)
            return 0;
        else
            return last - first + 1;
    }
    private int getFirstK(int [] array, int k, int low, int high){
        int mid = 0;
        while(low <= high){
            mid = low + (high -low) / 2;
            if(array[mid] == k){
                if(mid > 0 && array[mid - 1] != k || mid == 0)
                    return mid;
                else
                    high = mid - 1;
            }
            else if(array[mid] > k)
                high = mid - 1;
            else
                low = mid + 1;
        }
        return -1;
    }
    private int getLastK(int [] array, int k, int low, int high){
        int mid = 0;
        while(low <= high){
            mid = low + (high -low) / 2;
            if(array[mid] == k){
                if(mid < array.length - 1 && array[mid + 1] != k || mid == array.length - 1)
                    return mid;
                else
                    low = mid + 1;
            }
            else if(array[mid] > k)
                high = mid - 1;
            else
                low = mid + 1;
        }
        return -1;
    }
}