1. 程式人生 > >劍指Offer-陣列-(11)

劍指Offer-陣列-(11)

知識點/資料結構:陣列 flag:這道題目一定要記!!!

題目描述
統計一個數字在排序陣列中出現的次數。

思路:方法1

看到題目一種最簡單的解法就是把整個陣列遍歷一下

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

思路:方法2

因為陣列是排序的,所以優先想到了二分查詢。對於本題可以找到要找字母第一個出現的索引和最後一個出現的索引。然後相減加1就是出現的次數。

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
        int length = array.length;
        if(length == 0){
            return 0;
        }
        int firstK = getFirstK(array, k, 0, length-1);
        int lastK = getLastK(array, k, 0, length-1);
        if(firstK != -1 && lastK != -1){
             return lastK - firstK + 1;
        }
        return 0;
    }
    //遞迴寫法
    private int getFirstK(int [] array , int k, int start, int end){
        if(start > end){
            return -1;
        }
        int mid = (start + end)/2;
        if(array[mid] > k){
            return getFirstK(array, k, start, mid-1);
        }else if (array[mid] < k){
            return getFirstK(array, k, mid+1, end);
        }else if(mid-1 >=0 && array[mid-1] == k){//向前平移一個。
            return getFirstK(array, k, start, mid-1);
        }else{
            return mid;
        }
    }
    //迴圈寫法
    private int getLastK(int [] array , int k, int start, int end){
        int length = array.length;
        int mid = (start + end) /2;
        while(start <= end){
            if(array[mid] > k){
                end = mid-1;
            }else if(array[mid] < k){
                start = mid+1;
            }else if(mid+1 < length && array[mid+1] == k){
                start = mid+1;
            }else{
                return mid;
            }
            mid = (start + end)/2;
        }
        return -1;
    }
}