二分法:查找区间search for a range

问题描述:
给定一个排序数组nums(nums中有重复元素)与目标值target,如果 target在nums里出现,则返回target所在区间的左右端点下标,[左端点, 右端点],如果target在nums里未出现,则返回[-1, -1]。

例如:
arr = [2,3,4,4,4],target = 4,最终结果为[2,4]
arr = [2,3,4,5,6],target = 4,最终结果为[2,2]

分析:
最终的结果为一个区间,区间的话肯定有左端点,右端点。

题目是给定了一个有序数组,那么我们的目标就是确认左端点和右端点分别是多少即可。左端点即左侧再有没有比左端点更小的数值了,右端点即右侧再也没有比右端点更大小数值了

确认左端点的过程如下:
arr = [2,3,4,4,4],target = 4,那么需要走一遍二分查找的整个过程
如果我们发现arr[mid] == target,这个时候仍然需要进行判断,即mid == 0 或者arr[mid-1] < target,那么才能够认为此时 left = mid;即此时arr[mid]的左侧已经没有和target相等的节点了。

确认右端点的过程如下:
arr = [2,3,4,4,5],target = 4,那么同样需要走一遍二分查找的整个过程
如果我们发现arr[mid] == target,这个时候仍然需要进行判断,即mid == arr.size()-1 或者arr[mid+1] > target,那么才能够认为此时 right = mid;即此时arr[mid]的右侧已经没有和target相等的节点了。

以上过程实现如下:

pair<int,int> get_target_range(std::vector<int> &arr, int target) {
	int begin = 0;
	int end = arr.size() - 1;
	pair<int,int> range;
	range.first = -1;
	range.second = -1;
	/*计算左边界*/
	while(begin <= end) {
	    int mid = (begin + end) / 2;
	    if (arr[mid] == target) {
	    	if (mid == 0 || arr[mid - 1] < target) {//arr[mid]左侧没有比target更小的
	    		range.first = mid;
	    		break;
	    	} 
	    	end = mid - 1; //控制左侧的遍历
	    } else if (arr[mid] > target) {
	    	end = mid - 1;
	    } else {
	    	begin = mid + 1;
	    }
	}
	
	/*计算右边界*/
    begin = 0;
	end = arr.size() - 1;
	while(begin <= end) {
	    int mid = (begin + end) / 2;
	    if (arr[mid] == target) {
	    	/*arr[mid]右侧没有比target更大的*/
	    	if (mid == arr.size() - 1 || arr[mid + 1] > target) {
	    		range.second = mid;
	    		break;
	    	} 
	    	begin = mid + 1;
	    } else if (arr[mid] > target) {
	    	end = mid - 1;
	    } else {
	    	begin = mid + 1;
	    }
	}

	return range;
}

测试代码如下:

#include <iostream>
#include <vector>

using namespace std;
pair<int,int> get_target_range(std::vector<int> &arr, int target) {
	int begin = 0;
	int end = arr.size() - 1;
	pair<int,int> range;
	range.first = -1;
	range.second = -1;

	while(begin <= end) {
	    int mid = (begin + end) / 2;
	    if (arr[mid] == target) {
	    	if (mid == 0 || arr[mid - 1] < target) {
	    		range.first = mid;
	    		break;
	    	} 
	    	end = mid - 1;
	    } else if (arr[mid] > target) {
	    	end = mid - 1;
	    } else {
	    	begin = mid + 1;
	    }
	}

    begin = 0;
	end = arr.size() - 1;
	while(begin <= end) {
	    int mid = (begin + end) / 2;
	    if (arr[mid] == target) {
	    	if (mid == arr.size() - 1 || arr[mid + 1] > target) {
	    		range.second = mid;
	    		break;
	    	} 
	    	begin = mid + 1;
	    } else if (arr[mid] > target) {
	    	end = mid - 1;
	    } else {
	    	begin = mid + 1;
	    }
	}

	return range;
}

int main() {
	std::vector<int> arr1;
	int n;
	cin >> n;
	for (int i = 0;i < n; ++i) {
		int tmp;
		cin >> tmp;
		arr1.push_back(tmp);
	}
	int target;
	cin >> target;
	cout << "[" << get_target_range(arr1, target).first << "," << get_target_range(arr1, target).second << "]"; 
	return 0;
}

输出如下:

#输入
5
2 3 4 4 4
#输入target
4
#输出
[2,4]

#输入
5
4 4 4 4 4
#输入target
4
#输出
[0,4]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值