二分查找---按需求返回目标位置(相同元素时任意)、不大于目标的最大下标位置、不小于目标的最小下标位置

如题。二分查找简单快捷,但也可以多种变化,其中按返回值可以将其分为如题中的三种。简单实现+测试如下。
STL中的lower_bound和upper_bound实现类型功能,参考带源码:https://blog.csdn.net/qq_36172505/article/details/77380559
使用lower_bound(begin,end,greater)可以实现在降序数组二分查找。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

// 对相同元素,返回任意位置
// 对不存在的查找,返回任意位置(或-1)
int bs1(int target, vector<int> nums) {
	int L = 0, R = nums.size() - 1;
	while (L <= R) {
		int m = L + (R - L) / 2;
		if (nums[m] < target) {
			L = m + 1;
		}
		else if (nums[m] > target) {
			R = m - 1;
		}
		else {
			return m;
		}
	}

	return -1;
}

// 对相同元素,返回下标最小位置
// 对不存在的查找,返回不大于target的最大位置
int bs2(int target, vector<int> nums) {
	int L = 0, R = nums.size() - 1;
	while (L <= R) {
		int m = L + (R - L) / 2;
		if (nums[m] < target) {
			L = m + 1;
		}
		else if (nums[m] >= target) {
			R = m - 1;
		}
	}

	if (nums[R+1] == target)
		++R;
	return R;
}

// 对相同元素,返回下标最大位置
// 对不存在的查找,返回不小于target的最小位置
int bs3(int target, vector<int> nums) {
	int L = 0, R = nums.size() - 1;
	while (L <= R) {
		int m = L + (R - L) / 2;
		if (nums[m] <= target) {
			L = m + 1;
		}
		else if (nums[m] > target) {
			R = m - 1;
		}
	}

	if (nums[R] != target)
		++R;
	return R;
}

void Print(const vector<int>& nums) {
	for (int i = 0, size = nums.size(); i < size; ++i)
		cout << nums[i] << " ";
	cout << endl;
}

void PrintN(int index, const vector<int>& nums) {
	for (int i = 0; i < index; ++i)
			cout << "  ";
	cout << index << endl;
}

int main() {
	vector<int> nums1{1,2,4,5,5,5,5,6};//奇数
	vector<int> nums2{ 1,2,2,4,5,5,5,5,5,6,8,8 };//偶数

	cout << "一般查找:" << endl;
	//成功查找
	cout << "成功查找:查找5" << endl;
	Print(nums1);
	PrintN(bs1(5,nums1), nums1);
	Print(nums2);
	PrintN(bs1(5, nums2), nums2);
	//失败查找
	cout << "失败查找:查找3" << endl;
	Print(nums1);
	PrintN(bs1(3, nums1), nums1);
	Print(nums2);
	PrintN(bs1(3, nums2), nums2);
	cout << endl;

	cout << "返回不大于目标的最大下标---查找:" << endl;
	//成功查找
	cout << "成功查找:查找5" << endl;
	Print(nums1);
	PrintN(bs2(5, nums1), nums1);
	Print(nums2);
	PrintN(bs2(5, nums2), nums2);
	//失败查找
	cout << "失败查找:查找3" << endl;
	Print(nums1);
	PrintN(bs2(3, nums1), nums1);
	Print(nums2);
	PrintN(bs2(3, nums2), nums2);
	cout << endl;

	cout << "返回不小于目标的最小下标---查找:" << endl;
	//成功查找
	cout << "成功查找:查找5" << endl;
	Print(nums1);
	PrintN(bs3(5, nums1), nums1);
	Print(nums2);
	PrintN(bs3(5, nums2), nums2);
	//失败查找
	cout << "失败查找:查找3" << endl;
	Print(nums1);
	PrintN(bs3(3, nums1), nums1);
	Print(nums2);
	PrintN(bs3(3, nums2), nums2);
	cout << endl;

	return 0;
}

在这里插入图片描述

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值