在排序数组中查找数字Ⅰ

一、需求

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

二、暴力法

2.1  思路分析

1. 遍历整个数组，利用变量count累计目标值出现的次数，并返回；

2.2  代码实现

class Solution {
    public int search(int[] nums, int target) {
        int count = 0;
        for(int i = 0; i < nums.length; i++) {
            if(nums[i] == target) {
                count++;
            }
        }
        return count;
    }
}

2.3  复杂度分析

• 时间复杂度为O(n)；
• 空间复杂度为O(1)；

三、二分法

3.1  思路分析

1. 因为是有序数组，可通过二分法来降低时间复杂度；
2. 利用二分法查找目标元素的左边界和右边界，比如nums={5 7 7 8 8 10}，假设target = 8，那么其左边界下标为2，右边界下标为5，那么target个数为5 - 2 - 1，即right - left - 1；
3. 关于左、右边界的查找算法可以通过在本子上对给定数组进行模拟，比如先找5，然后找7、8，逐步得出正确的条件；
4. 这里有个提高效率的地方，就是找到右边界 i 后，此时 j = i - 1，若 j >= 0，判断 nums[j] 与 target 的关系，若nums[j] != target，说明当前nums[i-1] != target，又因为nums[i]作为右边界也不等于target，所以说明target不存在，就直接返回0；

3.2  代码实现

class Solution {
    public int search(int[] nums, int target) {
        int i = 0;
        int j = nums.length - 1;
        //查找右边界
        while(i <= j) {
            int mid = i + (j - i) / 2;
            if(nums[mid] <= target) {
                i = mid + 1;
            } else {
                j = mid - 1;
            }
        }
        int right = i;
        //找到右边界时，此时的j=i-1，那么就有如下判断
        if(j >= 0 && nums[j] != target) return 0;
        //查找左边界
        i = 0;
        j = nums.length - 1;
        while(i <= j) {
            int mid = i + (j - i) / 2;
            if(nums[mid] < target) {
                i = mid + 1;
            } else {
                j = mid - 1;
            }
        }
        int left = j;
        return right - left - 1;
    }
}

3.3  复杂度分析

• 时间复杂度为，二分查找的时间复杂度为；
• 空间复杂度为O(1)；

四、二分法优化

4.1  思路分析

1. 上述代码两轮二分查找冗余，在这里我们转换思路，本来是通过查找左、右边界来求个数；
2. 现在我们只查找右边界，找到target的右边界，然后找到target - 1的右边界，然后返回值相减就是要求的个数；

4.2  代码实现

class Solution {
    public int search(int[] nums, int target) {
        return helper(nums,target) - helper(nums,target - 1);
    }
    //查找目标元素的右边界
    public int helper(int[] nums, int target) {
        int i = 0;
        int j = nums.length - 1;
        while(i <= j) {
            int m = i + (j - i) / 2;
            if(nums[m] <= target) {
                i = m + 1;
            } else {
                j = m - 1;
            }
        }
        return i;
    }
}

五、学习地址

作者：Krahets

链接：https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof/solution/mian-shi-ti-53-i-zai-pai-xu-shu-zu-zhong-cha-zha-5/