LeetCode34
题目简述
在排序数组中查找元素的第一个和最后一个位置
难度:中等
描述:
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
题解
O(n)解法
虽然题目强调了算法复杂为O(log n),但是我在第一时间所想到的最优解法就只有O(n)算法。。。
确实,是我太菜(狗头)
当然,如果用O(n^2)甚至更复杂的算法实现也可以,思路也更自然
不过这样显然没有任何意义了
接下来陈述一下线性复杂度算法的实现
如果采用检索到target的出现后再建立一个新的循环找出target的结尾,那么是一个不可避免的O(n^2)算法,并且还会存在下标越界等其他的麻烦
这里我使用了双指针法,将i置于开头,j置于结尾
当然了使用迭代器iterator也可
同时移动下标,并进行条件判断:
cpp实现:
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> result(2, -1);
if(nums.empty()) return result;
int i = 0, j = nums.size() - 1;
while(j >= i && i < nums.size() && j < nums.size()) {
if(nums[i] == target) result[0] = i;
if(nums[j] == target) result[1] = j;
if(nums[i] < target) i++;
if(nums[j] > target) j--;
if(result[0] > -1 && result[1] > -1) break;
}
return result;
}
};
O(log n)解法
花了几分钟做出了O(n)解法以后——
博主突然想起来这题要求O(log n)的复杂度(裂开)
于是我又花了若干时间尝试该复杂度对应的解法
有一说一,
可能因为太菜了,我还是花了一段时间思考合适的解法的(彻底裂开)
咳咳,下面进入正题:
说到O(log n)算法,最经典的一个算法之一就是折半查找了
随着样本的线性增加,其时间消耗增加只呈现对数型增长
这题的本质其实就是一个稍加修改的折半查找算法
无非与只查找一个目标值不同,这题需要找到一个范围
而这个范围是一个闭区间 [x, y] ,或者是一个 ∅
那么我们只需要将题目简化为依次找到 左端点x 和 右端点y 就可以了
以下便是整个算法的执行思路:
- 判断传入向量是否为空
- 设置变量 left 和 right
- 二分查找定位 left 的位置
- 检测是否应返回 {-1, -1}
- 将 left 保存至返回的结果 result 中
- 再次二分查找定位 right 的位置
- 将 right 也保存至 result 中
- 返回 result
cpp实现:
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> result(2, -1);
if(nums.empty()) return result;
int len = nums.size();
int left = 0, right = len - 1;
while(left < right) {
int mid = left + (right-left)/2;
if(nums[mid] >= target) right = mid;
else left = mid + 1;
}
if(nums[left] != target) return result;
result[0] = left;
right = len;
while(left < right) {
int mid = left + (right-left)/2;
if(nums[mid] <= target) left = mid + 1;
else right = mid;
}
result[1] = right - 1;
return result;
}
};
STL实现(纯脑洞)
再次强调,纯脑洞解法,不具有任何普适性
从练习算法实现的角度也不建议使用
这里申明一下,关键字auto自动检测变量类型的用法始于c11,
这代表如果你使用的是一些尚未更新的上古编译器(比如我的codeblocks),是无法正常编译的
cpp实现:
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
auto left = lower_bound(nums.begin(), nums.end(), target);
auto right = upper_bound(nums.begin(), nums.end(), target);
if(left == right) return {-1,-1};
return {left-nums.begin(), right-nums.begin()-1};
}
}
以上便是我关于LeetCode34的全部解法