leetcode34——Search for a Range

题目大意:在给出的升序数组中找出目标数字的起始和终止下标

分析:二分搜索。不过发现普通遍历也挺快的。

代码:

方法一:普通遍历

class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> ans;
int start = -1, end = -1;
for (int i = 0;i < nums.size();i++) {
if (nums[i] == target) {
end = i;
if (start == -1)
start = i;
}
else if (nums[i] > target) {
ans.push_back(start);
ans.push_back(end);
return ans;
}
}
ans.push_back(start);
ans.push_back(end);
return ans;
}

};

方法二:二分搜索。

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int left = 0,right = nums.size() - 1;
        int start = -1,end = -1;
        vector<int> ans;
        while(left <= right){
            int mid = (left + right) / 2;
            if(nums[mid] < target){
                left = mid + 1;
            }
            else if(nums[mid] > target){
                right = mid - 1;
            }
            else{
                int i = mid - 1;
                while(i >= 0 && nums[i] == target){
                    i--;
                }
                start = i + 1;
                i = mid + 1;
                while(i < nums.size() && nums[i] == target){
                    i++;
                }
                end = i - 1;
                ans.push_back(start);
                ans.push_back(end);
                return ans;
            }
        }
        ans.push_back(start);
        ans.push_back(end);
        return ans;
    }
};

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页