在排序数组中查找数字Ⅰ

最新推荐文章于 2022-05-29 13:23:41 发布

皮卡qiu~

最新推荐文章于 2022-05-29 13:23:41 发布

阅读量172

点赞数

分类专栏：剑指Offer 文章标签：二分法算法 leetcode

本文链接：https://blog.csdn.net/Sruggle/article/details/110350982

版权

剑指Offer 专栏收录该内容

73 篇文章 1 订阅

订阅专栏

一、需求

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

二、暴力法

2.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 思路分析

因为是有序数组，可通过二分法来降低时间复杂度；
利用二分法查找目标元素的左边界和右边界，比如nums={5 7 7 8 8 10}，假设target = 8，那么其左边界下标为2，右边界下标为5，那么target个数为5 - 2 - 1，即right - left - 1；
关于左、右边界的查找算法可以通过在本子上对给定数组进行模拟，比如先找5，然后找7、8，逐步得出正确的条件；
这里有个提高效率的地方，就是找到右边界 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 思路分析

上述代码两轮二分查找冗余，在这里我们转换思路，本来是通过查找左、右边界来求个数；
现在我们只查找右边界，找到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;
    }
}