剑指Offer 统计一个数字K在排序数组中出现的次数 Java

题目描述

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

1.暴力

1. 当前是不是==k是count++ 不是break
2. 这种方法的话，平均时间复杂度是O(n)

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
         int count=0;
        for(int i=0;i<array.length;i++){
            if(array[i]==k)
                count++;
            if(array[i]>k)
                break;
        }
        return count;
    }
}

2.二分查找

1. 有序数组，考虑二分查找，查到第一个K的index1 最后一个的 index2 returnindex2-index1+1
2. 找第一个K 二分mid的值1. >k k在左边 2. <k 在右边 3. =k判断前一个是不是K是的话说明还是要左查找 不是当前就是第一个
3. 找最后一个 思路和上边基本类似
4. 这种时间复杂度logn

public class Solution {
    public int GetNumberOfK(int [] arr , int k) {
        if(arr==null||arr.length==0)
            return 0;
         int index1=getFirst(arr,0,arr.length-1,k);
        int index2=getLast(arr,0,arr.length-1,k);
        if(index1>-1&&index2>-1)
            return index2-index1+1;
        return 0;
    }
    public  int getFirst(int arr[],int start,int end,int k){
        if(start>end)
            return -1;
        int mid=start+(end-start)/2;
        if(k>arr[mid]){
            start=mid+1;
        }
        if(k<arr[mid]){
            end=mid-1;
        }
        if(k==arr[mid]){
           if((mid>=1&&arr[mid-1]!=k)||mid==0)//注意不要越界
                return mid;
            else
                end=mid-1;
        }
        return getFirst(arr,start,end,k);
    }
    public  int getLast(int arr[],int start,int end,int k){
        if(start>end)
            return -1;
        int mid=start+(end-start)/2;
        if(k>arr[mid]){
            start=mid+1;
        }
        if(k<arr[mid]){
            end=mid-1;
        }
        if(k==arr[mid]){
            if((mid<=arr.length-2&&arr[mid+1]!=k)||mid==arr.length-1)//注意不要越界
                return mid;
            else
                start=mid+1;
        }
        return getLast(arr,start,end,k);
    }
}