剑指 Offer 53 - I. 在排序数组中查找数字 I

好感好感

于 2023-02-06 14:33:45 发布

阅读量80

点赞数

分类专栏：剑指Offer 文章标签：算法性能优化

本文链接：https://blog.csdn.net/m0_56368698/article/details/128901018

版权

剑指Offer 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

剑指 Offer 53 - I. 在排序数组中查找数字 I

解法一：简单遍历

一次遍历统计即可。

时间复杂度：O(n) ，空间复杂度：O(1)。

int search(vector<int>& nums, int target) {
        int count=0;
        for(int num:nums)
            if(num==target) count++;
        return count;
    }

解法二：二分查找

两次二分分别确定target的左边界、右边界。

时间复杂度：O(logn) ，空间复杂度：O(1)。

int search(vector<int>& nums, int target) {
        //数组为空，直接返回0
        if(nums.size()==0) return 0;

        int left = 0, right = 0;
        // 第一次二分找左边界
        int l=0,r=nums.size()-1;
        while(l<=r){
            int mid = (l+r)>>1;
            if(nums[mid]>=target){ //！！！
                r = mid - 1;
                left = mid;
            }
            else l = mid + 1;
        }
        //判断是否真的找到，找不到直接返回
        if(nums[left]!=target) return 0;

        // 第二次二分找右边界
        l=0,r=nums.size()-1;
        while(l<=r){
            int mid = (l+r)>>1;
            if(nums[mid]<=target){ //！！！
                l = mid + 1; 
                right = mid;
            }
            else r = mid - 1;
        }

        return right - left + 1;
    }

解法三：二分查找代码优化

解法二代码较繁杂，还有各种判断。可以转换一下思路：

定义一个函数bin：二分寻找第一个大于等于target的元素位置（若不存在，返回的是数组右边界）。

然后两次调用相减： bin(nums,target+1) - bin(nums,target); 即为答案。

时间复杂度：O(logn) ，空间复杂度：O(1)。

    int search(vector<int>& nums, int target) {
        return bin(nums,target+1) - bin(nums,target);
    }
    //二分寻找第一个大于等于target的元素位置（若不存在，返回的是数组右边界）
    int bin(vector<int>& nums, int target){
        int l=0,r=nums.size()-1;
        while(l<=r){
            int mid = (l+r)>>1;
            if(nums[mid]>=target) r = mid - 1;
            else l = mid + 1;
        }
        return l;
    }