二分查找(个人笔记)

前言

在学习MOOC数据结构(陈越)的有关"树"的部分时, 作为前导知识, 同时也是一直不是很懂实现方法的二分查找, 今天借助洛谷题单, 简单加深了一下对二分查找的理解以及C++相关库的使用方法, 特此记录

引例1

洛谷P2249【深基13.例1】查找

分析

给出一组排好序的数列, 询问其中元素位置. 数据范围较大, 普通查找方式会TLE. 由于是排好序的数列, 所以使用二分查找即可.

另外本题与最最最最最基础二分查找也有些许不同, 因为题目中所需要的是所查询的序列中最前面的元素的地址(用lower_bound可以轻松解决, 但这里为了理解原理, 仍手写代码). 所以在代码上有所修改.

代码实现

#include<cstdio>
const int MAXN = 1000000+10;
int a[MAXN];
int n, m;

int Find(int ans)
{
	int left = 1, right = n;
	int mid = (left+right)/2;
	
	while(left<right)
	{
		if(ans<=a[mid]) //当所查找元素与当前位置元素相同时,令右坐标等于当前位置,进而寻找最前面的元素
		{
			right = mid;
		}
		else
		{
			left = mid + 1;
		}
		mid = (left+right)/2;
	}
	if(a[right]==ans)	//退出循环时l==r, r位置即为最后确定下来的位置
	{
		return right;
	}
	else
		return -1;
}

int main(void)
{
	scanf("%d %d", &n, &m);
	for(int i=1; i<=n; i++)
	{
		scanf("%d", &a[i]);
	}
	for(int i=1; i<=m; i++)
	{
		int x;
		scanf("%d", &x);
		int loc = Find(x);
		
		if(i==1)	//注意输出格式
			printf("%d", loc);
		else
			printf(" %d", loc);
		
	}
	
	return 0;
}

引例二

洛谷 P1102 A-B 数对

分析

给出一组数列和一个数C, 问数列中有多少个数对A,B(A和B不相等), 使得A-B=C.

本题似乎与二分查找没什么关系, 但可以用C++的两个内置函数,lower_bound和upper_bound(这两个函数据说是用二分查找的原理写的), 他们的使用形式如下:
lower_bound(查找起始位置, 查找结束位置, 所查找元素), upper_bound同理.

这两个函数需要包含头文件#include<algorithm> 以及std空间

他们的返回值分别为所查找元素第一次出现的 地址值, 和第一次不同时(即最后一个元素下一项)的 地址值.

因此如果想使用这两个函数得到所求元素的下标, 需要在函数后减去"查找起始位置"的地址值.

代码实现

#include<cstdio>
#include<algorithm>
using namespace std;

const int MAXN = 200010;
long long a[MAXN];
long long n, c, ans;
//struct number{
//	int x;
//	int num;
//}b[MAXN];


int main(void)
{
	scanf("%lld %lld", &n, &c);
	for(int i=0; i<n; i++)
	{
		scanf("%lld", &a[i]);
	}
	sort(a, a+n);
	
	for(int i=0; i<n; i++)
	{
		long long *t = upper_bound(a,a+n,a[i]+c);
		printf("%p\n", t);
		printf("%lld\n\n", t);
		// 下条语句为关键用法的体现.
		ans += ((upper_bound(a,a+n,a[i]+c)-a)-(lower_bound(a,a+n,a[i]+c)-a));
	}
	printf("%lld", ans);
	
	return 0;
}