[剑指offer]在排序数组中查找数字I

[剑指offer]在排序数组中查找数字I

剑指offer-在排序数组中查找数字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

解题思路

• 面试官出这个题大概率是想看你写二分
• 用两次二分，分别找这个数字出现的右边界和左边界。

实现代码

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int len=nums.size();
        if(len==0)
            return 0;
        int count=0;
        int i=0,j=len-1;
        while(i<=j){//第一次二分查找，找右边界
            int mid=(i+j)/2;
            if(nums[mid]<=target)
                i=mid+1;
            else
                j=mid-1;
        }
        if(j>0&&nums[j]<target)//数组中没有target，可以提前出循环
            return 0;
        int right=j;
        i=0;
        while(i<=j){//第二次二分查找，找左边界
            int mid=(i+j)/2;
            if(nums[mid]<target)
                i=mid+1;
            else
                j=mid-1;
        }
        int left=i;
        return right-left+1;
    }
};