欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【数组】剑指 Offer 53 - I. 在排序数组中查找数字 I(简单)

程序员文章站 2023-12-28 13:32:58
...

【题目】
统计一个数字在排序数组中出现的次数。
【示例 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) {
        map<int,int> m;
        for(auto x:nums)
            m[x]++;
        return m[target];
    }
};

【一行代码】

class Solution {
public:
    int search(vector<int>& nums, int target) {
        return upper_bound(nums.begin(),nums.end(),target)-lower_bound(nums.begin(),nums.end(),target);
    }
};

【二分法】

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int l=0,r=nums.size()-1,m=(l+r)/2,cnt=0;
        if(r==-1)
            return 0;
        while(l<=r){
            m=(l+r)/2;
            if(nums[m]==target)
                break;
            else if(nums[m]>target){
                r=m-1;
            }else
                l=m+1;
        }
        if(nums[m]==target){
            int i=m;
            while(i<nums.size()&&nums[i]==target){
                    cnt++;
                    i++;
            }
            i=m-1;
            while(i>=0&&nums[i]==target){
                    cnt++;
                    i--;
            }
        }
        return cnt;
    }
};

【二分法简化版】

class Solution {
public:
    int binarySearch(vector<int> nums,int target){
        int i = 0, j = nums.size() - 1;
        while(i <= j) {
            int m = (i + j) / 2;
            if(nums[m] <= target) i = m + 1;
            else j = m - 1;
        }
        return i;
    }
    int search(vector<int>& nums, int target) {
        return binarySearch(nums, target) - binarySearch(nums, target - 1);
    }
};

上一篇:

下一篇: