leetcode刷题笔记 34.在排序数组中查找元素的第一个和最后一个位置【中等】

最新推荐文章于 2024-03-15 22:12:01 发布

EGNIR

最新推荐文章于 2024-03-15 22:12:01 发布

阅读量107

点赞数

分类专栏： leetcode刷题笔记文章标签： leetcode 二分法

本文链接：https://blog.csdn.net/weixin_46477226/article/details/119152009

版权

leetcode刷题笔记专栏收录该内容

104 篇文章 0 订阅

订阅专栏

两次二分法

vector<int> searchRange(vector<int>& nums, int target) {
	int n = nums.size();
	int l = 0, r = n - 1;
	vector<int> ans = { -1,-1 };
	if (n == 0) {
		return ans;
	}
	int mid;
	while (l < r) {
		mid = l + (r - l) / 2;
		if (nums[mid] < target) {
			l = mid + 1;
		}
		else {
			r = mid;
		}
	}
	if (nums[l] == target) {
		ans[0] = l;
	}
	l = 0; r = n - 1;
	while (l < r) {
		mid = l + (r - l) / 2;
		if (nums[mid] > target) {
			r = mid;
		}
		else {
			l = mid + 1;
		}
	}
	if (l == n - 1 && nums[l] == target) {
		ans[1] = l;
	}
	else if (l - 1 >= 0 && nums[l - 1] == target) {
			ans[1] = l - 1;
		}
	return ans;
}