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

最新推荐文章于 2024-10-06 16:32:56 发布

坚持健身的程序员

最新推荐文章于 2024-10-06 16:32:56 发布

阅读量100

点赞数

文章标签： leetcode 算法

本文链接：https://blog.csdn.net/hou9021/article/details/115433953

版权

二分法排序数组目标值计数 C++代码

关键词由CSDN通过智能技术生成

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

示例 1:

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

示例 2:

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

方法：二分法
首先由经典的二分法搜索得到nums中任意一个target的位置，之后分别向两边计数。
特殊情况：

$n u m s = []$ ，此时 $r = - 1 ， l > r$ ，直接跳到最下 $r e t u r n 0$ 。
只有一项 $n u m s [1] ， t a r g e t = 1$ ，此时第一次进入 $w h i l e$ 循环， $n u m s [m]$ 就是 $= t a r g e t$ 的，直接开始计数，在计数中， $m_l = -1，m_r = 1$ 均超出索引范围，都不满足计数 $w h i l e$ 的条件，因此直接输出此时的 $n = 1$ 。

C++

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int r = nums.size()-1, l = 0, n = 1;

        int m = 0;
        while(l<=r){
            m = (l+r)/2;
            if(nums[m]<target) l = m+1;
            else if(nums[m]>target) r = m-1;
            else{
                    int m_l = m-1, m_r = m+1;
                    while(m_l >= 0 && nums[m_l] == target) {
                        n++;
                        m_l--;
                    }
                    while(m_r<=nums.size()-1 && nums[m_r] == target) {
                        n++;
                        m_r++;
                    }
                    return n;    
            }
        }
        return 0;

    }
};