力扣 题目34-- 在排序数组中查找元素的第一个和最后一个位置

题目

---

题解

---

 

改造二分搜索 具体说明在代码注释#

代码

---

#include<iostream>
#include<vector>
using namespace std;
vector<int> ersearch(vector<int>& nums, int target, int left, int right)
{
    vector<int>result;
    while (left <= right) {
        int middle = left + ((right - left) / 2);
        if (nums[middle] > target) {
            right = middle - 1;
        }
        else if (nums[middle] < target) {
            left = middle + 1;
        }
        else {
            //这里找到其中一个元素 但是他不一定是最后一个或者是第一个
            //备份一下 
            int backups = middle;
            //判断一下他的上一个是target？如果是则更新 这样就能找到最小
            while (middle != 0 && nums[middle] == nums[middle - 1]) {
                middle--;
            }
            //放入最小
            result.push_back(middle);
            //判断一下他的下一个是target？如果是则更新 这样就能找到最大
            while (backups != nums.size()-1 && nums[backups] == nums[backups + 1]) {
                backups++;
            }
            //放入最大
            result.push_back(backups);
            return result;
        }
    }
    //如果没有 就返回-1 -1
    return {-1,-1 };
}

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        //调用二分搜索
        return ersearch(nums, target, 0, nums.size()-1);
    }
};


int main() {
    Solution sol;
    vector<int> nums = { 5,7,7,8,8,10 };
    int target = 5;
    vector<int> result=sol.searchRange(nums, target);
    for (int i = 0; i < result.size();i++) {
        cout << result[i] << endl;;
    }
}