【LeetCode】34_Search for a Range

来源:互联网 发布:电力仿真软件多少钱 编辑:程序博客网 时间:2024/06/05 11:31

题目

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].














解析

题目很简单,就是普通的查找,主要是要求时间复杂度O(log n),所以我想到的方法就是二分查找法。

与一般二分查找不同的是,这个序列有可能出现重复的的数字,我的解决方法是二分法,然后如果中位数正好等于target了,就往前和往后搜索即可。

最后再针对几种可能出现的情况分类讨论一下就好了。

代码如下

class Solution {public:void serachMy(vector<int>& nums, int start, int end, int target,vector<int>& ss){if (start == end){if (nums[start] == target){ss.push_back(start);}return;}int mid = (end - start)/2+start;if (nums[mid] < target){serachMy(nums, mid+1>nums.size()?mid:mid+1, end, target,ss);}else if(nums[mid] > target){serachMy(nums, start, mid-1<0?mid:mid-1, target,ss);}else{int flagEnd = mid;while (flagEnd<nums.size() && nums[flagEnd] == target){flagEnd++;}int flagStart = mid;while (flagStart>=0 && nums[flagStart] == target){flagStart--;}ss.push_back(flagStart+1);ss.push_back(flagEnd-1);return;}}    vector<int> searchRange(vector<int>& nums, int target) {       <span style="white-space:pre"></span>        vector<int> ss;vector<int> ret;serachMy(nums, 0, nums.size()-1,target,ss);if (ss.empty()){ret.push_back(-1);ret.push_back(-1);}else if (ss.size()==1){ret.push_back(ss[0]);ret.push_back(ss[0]);}else{ret.push_back(ss[0]);ret.push_back(ss.back());}return ret;    }};

方法比较中规中矩,主要是注意一下边界条件。

再看看大神的代码。

class Solution {public:vector<int> searchRange(int A[], int n, int target) {    const int l = distance(A, lower_bound(A, A + n, target));    const int u = distance(A, prev(upper_bound(A, A + n, target)));    if (A[l] != target) // not found        return vector<int> { -1, -1 };    else        return vector<int> { l, u };    }};
什么也不想说了,用了这么多STL的函数。来总结一下知识点。

1.back()函数

可以取vector的最后一个元素。如 nums.back()

2.lower_bound()

函数lower_bound()在first和last中的前闭后开区间进行二分查找,返回大于或等于val的第一个元素位置。如果所有元素都小于val,则返回last的位置,如果所有元素都小于val,则返回last的位置,且last的位置是越界的!!~

3.upper_bound()

函数upper_bound()返回的在前闭后开区间查找的关键字的上界,如一个数组number序列1,2,2,4.upper_bound(2)后,返回的位置是3(下标)也就是4所在的位置,同样,如果插入元素大于数组中全部元素,返回的是last。(注意:此时数组下标越界!!)

返回查找元素的最后一个可安插位置,也就是“元素值>查找值”的第一个元素的位置

4.distance

c++中,distance为一函数模版,返回两个迭代器间的距离







0 0
原创粉丝点击