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

题目描述

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

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: 2

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: 0

限制：

0 <= 数组长度 <= 50000

code

• 顺序查找
• 20ms

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int ans=0;
        for(int i=0; i<nums.size(); i++){
            if(nums[i]==target)
                ans++;
        }
        return ans;
    }
};

• 数组长度0-50000，降低时间复杂度，有序数组查找很容易想到二分查找
• 先二分找左边界再二分找右边界
• 20ms（应该是数据量不够大，看不出区别）

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int left=0,right=nums.size()-1;
        int leftMargin=left,rightMargin=right;
        //找到左边界
        while(left<right){
            int mid=left+((right-left)>>1);//左中位数
            if(nums[mid]<target){
                left=mid+1;
            }
            else {
                right=mid;
            }
        }
        leftMargin=left;
        //找到右边界
        left=0;
        right=nums.size()-1;
        while(left<right){
            int mid=left+((right-left+1)>>1);//注意这里选择右中位数,否则会进入死循环
            if(nums[mid]>target){
                right=mid-1;
            }
            else {
                left=mid;
            }
        }
        rightMargin=right;
        return rightMargin-leftMargin+1;
    }
};

补充

• 为什么选择右中位数mid=left+(right-left+1)/2？
  • LeetCode 题解 | 69. x 的平方根