LeetCode OJ:Search for a Range(区间查找)

Given a sorted array of integers, find the starting and ending position of a given target value.

Your algorithm's runtime complexity must be in the order of O(log n).

If the target is not found in the array, return [-1, -1].

For example,
Given [5, 7, 7, 8, 8, 10] and target value 8,
return [3, 4].

查找一个区间开头以及结尾的下标,分两次二分查找,一次向左一次向右即可,代码如下:

 1 class Solution {
 2 public:
 3     vector<int> searchRange(vector<int>& nums, int target) {
 4         int sz = nums.size();
 5         int left = bs(nums, 0, sz, target, true);
 6         int right = bs(nums, 0, sz, target, false);
 7         vector<int> res;
 8         res.push_back(left);
 9         res.push_back(right);
10         return res;
11     }
12 
13     int bs(vector<int>& nums, int beg, int end, int target, int goLeft)
14     {
15         if(beg > end)
16             return -1;
17         int mid = (beg + end)/2;
18         if(nums[mid] == target){
19             int tmpAns = (goLeft == true ? bs(nums, beg, mid - 1, target, goLeft) : bs(nums, mid + 1, end, target, goLeft));
20             return tmpAns == -1 ? mid : tmpAns;
21         }else if(nums[mid] < target){
22             return bs(nums, mid + 1, end, target, goLeft);
23         }else{
24             return bs(nums, beg, mid - 1, target, goLeft);
25         }
26     }
27 };

 

posted @ 2015-10-29 19:22  eversliver  阅读(294)  评论(0编辑  收藏  举报