【劍指offer-Java版】38數字在排序數組中出現的次數

數字在排序數組中出現的次數:
最簡單粗暴的方法是O(n^2)

可能的解法可以採用二分,首先根據二分找到給定數字在數組中的位置,然後再左右二分,找到邊界(第一個和最後一個),左右邊界的差值就是出現次數


    public class _Q38<T> {

    public int GetNumberOfK(int nums[], int k){
        if(nums == null) return 0;

        int first = GetFirstK(nums, k, 0, nums.length - 1);
        int last = GetLastK(nums, k, 0, nums.length - 1);

        int result = 0;
        if(first > -1 && last > -1){
            result = last - first + 1;
        }

        return result;
    }

    private int GetFirstK(int nums[], int k, int start, int end){
        if(nums == null || start > end) return -1;

        int mid = (start + end) / 2;
        if(nums[mid] == k){
            if(mid == 0) return mid;
            if(mid > 0 && nums[mid - 1] != k) return mid;
            return GetFirstK(nums, k, start, mid - 1); // 容易遺漏的邏輯分支
        }else if(nums[mid] > k){
            return GetFirstK(nums, k, start, mid - 1);
        }else if(nums[mid] < k){
            return GetFirstK(nums, k, mid + 1, end);
        }

        return -1;
    }

    private int GetLastK(int nums[], int k, int start, int end) {
        if (nums == null || start > end) return -1;

        int mid = (start + end) / 2;
        if (nums[mid] == k) {
            if (mid == end) return mid;
            if (mid < end && nums[mid + 1] != k) return mid;
            return GetLastK(nums, k, mid + 1, end); // 容易遺漏的邏輯分支
        } else if (nums[mid] > k) {
            return GetLastK(nums, k, start, mid - 1);
        } else if (nums[mid] < k) {
            return GetLastK(nums, k, mid + 1, end);
        }

        return -1;
    }

    }

測試代碼:


    public class _Q38Test extends TestCase {

    _Q38 numOfK = new _Q38();

    public void test(){

        int nums1[] = {1, 2, 3, 3, 3, 3, 4, 5};
        int k1 = 3;
        int k2 = 6;
        int k3 = 4;

        int result = 0; 
        result = numOfK.GetNumberOfK(nums1, k1);
        System.out.println(result);
        result = numOfK.GetNumberOfK(nums1, k2);
        System.out.println(result);
        result = numOfK.GetNumberOfK(nums1, k3);
        System.out.println(result);
        result = numOfK.GetNumberOfK(null, k3);
        System.out.println(result);


    }
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章