二分法模板大全

最近看了看刷了刷二分的题目,发现自己对于二分的理解还不够透彻,参考了其他大佬的资料再加上自己的总结,重新手写了一遍二分的模板,具体模板如下:

#include <bits/stdc++.h> 
using namespace std;

//返回小于等于num的第一个数字 
int bs_1(int arr[], int num, int l, int r){
	while (l < r){
		int mid = (l + r) / 2; 
		if (arr[mid] >= num)    
			r = mid;
		else
			l = mid + 1;
	} 
	return l; 
}

//返回大于等于num的最后一个数字 
int bs_2(int arr[], int num, int l, int r){
	while (l < r){
		int mid = (l + r + 1) / 2;  
		if (arr[mid] <= num)	//  返回大于等于num的最后一个数 
			l = mid;
		else
			r = mid - 1;     
	}
	return l; 
}
  
// r指向大于等于num的最后一个数   l指向大于num 的第一个数 
int bs_3(int arr[], int num, int l, int r){
	while (l <= r){
		int mid = (l + r) / 2;
		if (arr[mid] <= num)
			l = mid + 1;
		else
			r = mid - 1;
	}
	return r; 
}

// l指向小于等于num的第一个数   r指向小于num 的最后一个数 
int bs_4(int arr[], int num, int l, int r){
	while (l <= r){
		int mid = (l + r) / 2;
		if (arr[mid] >= num)
			r = mid - 1;
		else
			l = mid + 1;
	}
	return l; 
}

int main(){
	int arr1[] = {0, 8, 8, 8, 9};
	int arr2[] = {0, 5, 8, 8, 8, 9}; 
	int arr3[] = {0, 5, 8, 8, 8, 8};
	int arr4[] = {0, 1, 4, 5, 7, 9};
	
	cout << bs_1(arr1, 8, 0, 4)  << " " << bs_2(arr1, 8, 0, 4) <<  " " <<  bs_3(arr1, 8, 0, 4) << " " << bs_4(arr1, 8, 0, 4) << endl;
	cout << bs_1(arr2, 8, 0, 5)  << " " << bs_2(arr2, 8, 0, 5) <<  " " <<  bs_3(arr2, 8, 0, 5) << " " << bs_4(arr2, 8, 0, 5) << endl;
	cout << bs_1(arr3, 8, 0, 5)  << " " << bs_2(arr3, 8, 0, 5) <<  " " <<  bs_3(arr3, 8, 0, 5) << " " << bs_4(arr3, 8, 0, 5) << endl;
	cout << bs_1(arr4, 8, 0, 5)  << " " << bs_2(arr4, 8, 0, 5) <<  " " <<  bs_3(arr4, 8, 0, 5) << " " << bs_4(arr4, 8, 0, 5) << endl;
	

	return 0;
}

二分法模板大全

观察最后结果可以发现bs_1和bs_4得到的结果相同,bs_2和bs_3得到的结果相同,总体而言bs_3和bs_4的写法更加灵活,返回l和r分别可以得到不同的值。但两种写法都可以运用于处理重复元素的二分查找的问题,笔者更加习惯于用后两个函数,读者可选择1,2或3,4其一使用即可,使用时将if的条件修改为check函数即可。

上一篇:Python爬虫(二)导包、解释urllib、bs4


下一篇:Bootstrap5 多级dropdown