LCR 172.统计目标成绩的出现次数

十年一觉尘与土2023-10-14 14:05

题目来源：

leetcode题目，网址：LCR 172. 统计目标成绩的出现次数 - 力扣（LeetCode）

解题思路：

两次二分，一次寻找首次出现的位置，一次寻找最后一次出现的位置。

解题代码：

复制代码

class Solution {
public:
    int countTarget(vector<int>& scores, int target) {
        if(scores.size()==0){
            return 0;
        }else if(scores.size()==1){
            return scores[0]==target?1:0;
        }
        int left=getLeft(scores,target);
        if(left==-1){
            return 0;
        }
        int right=getRight(scores,target,left);
        return right-left+1;
    }
    int getRight(vector<int>& scores,int target,int left){
        if(scores[scores.size()-1]==target){
            return scores.size()-1;
        }
        int right=scores.size()-2;
        while(left<=right){
            int mid=left+(right-left)/2;
            if(scores[mid]==target && scores[mid+1]>target){
                return mid;
            }else if(scores[mid]<=target){
                left=mid+1;
            }else{
                right=mid-1;
            }
        }
        return -1;
    }
    int getLeft(vector<int>& scores,int target){
        if(scores[0]==target){
            return 0;
        }
        int left=1;
        int right=scores.size()-1;
        while(left<=right){
            int mid=left+(right-left)/2;
            if(scores[mid-1]<target &&  scores[mid]==target){
                return mid;
            }else if(scores[mid]>=target){
                right=mid-1;
            }else{
                left=mid+1;
            }
        }
        return -1;
    }
};

复制代码

总结：

官方题解也是两次二分，不过他利用变量将写两个二分函数合为一个。