题解 | #找出特定体重的牛群#
找出特定体重的牛群
https://www.nowcoder.com/practice/bbc4c61a5bb64f329663b300a634ae6a
大家好,我是开车的阿Q,自动驾驶的时代已经到来,没时间解释了,快和阿Q一起上车。作为自动驾驶系统工程师,必须要有最好的C++基础,让我们来一起刷题吧。
题目考察的知识点
这道题目同样考察二分查找算法的应用。
题目解答方法的文字分析
给定一个非递增的体重数组和一个目标体重 target
,要求找出给定体重的牛在牛棚中的起始位置和结束位置。由于要求时间复杂度为 O(log n),我们可以使用二分查找算法来解决。
思路步骤如下:
- 初始化两个指针
left
和right
分别指向数组的第一个和最后一个元素。 - 在每一轮二分查找中,计算中间元素的下标
mid
。 - 如果
weights[mid] == target
,说明找到了一个目标体重的牛,需要继续查找它的起始位置和结束位置。 - 初始化
start
和end
变量为mid
,然后向左查找起始位置,直至weights[start]
不等于target
。 - 向右查找结束位置,直至
weights[end]
不等于target
。 - 返回起始位置和结束位置,如果找不到,返回
{-1, -1}
。
本题解析所用的编程语言
本题解析所用的编程语言是C++。
完整且正确的编程代码
class Solution { public: vector<int> searchRange(vector<int>& weights, int target) { vector<int> result = {-1, -1}; // 初始化结果数组,起始位置和结束位置均为 -1 int left = 0, right = weights.size() - 1; while (left <= right) { int mid = left + (right - left) / 2; // 计算中间元素的下标 if (weights[mid] == target) { // 查找起始位置 int start = mid; while (start >= 0 && weights[start] == target) { start--; } // 查找结束位置 int end = mid; while (end < weights.size() && weights[end] == target) { end++; } result[0] = start + 1; // 设置起始位置 result[1] = end - 1; // 设置结束位置 return result; // 返回结果数组 } if (weights[mid] > target) { left = mid + 1; // 如果中间元素大于目标值,说明目标值在左半部分,将左指针移动到 mid + 1 } else { right = mid - 1; // 如果中间元素小于目标值,说明目标值在右半部分,将右指针移动到 mid - 1 } } return result; // 如果没有找到目标值,返回结果数组 } };
阿Q的题解 文章被收录于专栏
阿Q秋招刷过的题