【个人学习记录】数据结构——二分查找、upper_bound的使用（[ZCMU OJ]1679: 查找2 & 1680: 查找1）

首先列出二分查找法的模板（来自《数据结构与算法分析》）：

int BinarySearch(int A[ ],int X,int N)
{
	int Low,Mid,High;
	Low=0;High=N-1;
	while(Low<=High)
	{
		Mid=(Low+High)/2;
		if(A[Mid]<X)
		Low=Mid+1;
		else if(A[Mid]>X)
		High=Mid-1;
		else
		return Mid;//found
	}
	return -1;
}

---------------------------------------------------------------------------------------------------------------------------------

1680: 查找1

Description

给你一个长度是n的序列A，然后，有m次询问，每次询问是一个数字X，请你告诉我X在序列A中是否存在，存在输出YES，否则输出NO

Input

第一行 ，n,m,(n,m<=100000)

第二行n个数(每个数<=1000)

第三行m个数

Output

输出答案

Sample Input

5 4

2 5 4 3 5

2 5 8 9

Sample Output

YES

YES

NO

NO

本题可直接套用模板，若经过二分查找之后该数据存在于序列A中，则输出其位置。反之，则输出-1。本题易超时，用scanf和printf可节省时间。在二分查找之前应对序列进行排序。

#include<bits/stdc++.h>
using namespace std;
int BinarySearch(int a[],int x,int n)
{
	int high=n-1,low=0,mid;
	while(low<=high)
	{
		mid=(high+low)/2;
		if(x>a[mid])
		low=mid+1;
		else if(x<a[mid])
		high=mid-1;
		else
		return mid;
	}
	return -1;
}
int a[100010];
int main()
{
	int n,m;
	int temp;
	while(~scanf("%d %d",&n,&m))
	{
		for(int i=0;i<n;i++)
		scanf("%d",&a[i]);
		
		sort(a,a+n);
		
		for(int i=0;i<m;i++)
		{
			scanf("%d",&temp);
			if(BinarySearch(a,temp,n)!=-1)
			printf("YES\n");
			else
			printf("NO\n");
		}
	}
} 

---------------------------------------------------------------------------------------------------------------------------

1679: 查找2

Description

给你一个长度是n的序列A，然后，有m次询问，每次询问是一个数字X，请你告诉我X在序列A中有多少个数是不大于它的

Input

第一行 ，n,m,(n,m<=100000)

第二行n个数(每个数<=1000)

第三行m个数

Output

输出答案

Sample Input

5 4

2 5 4 3 5

2 5 8 9

Sample Output

1

5

5

5

首先了解函数upper_bound：对于已经排好序的序列，在左闭右开的区间内进行二分查找并且返回大于或等于指定元素大小第一个元素的位置。

 例如：int i = upper_bound(a, a+ n, num) - a 表示：从小到大，num最多能插入到数组a的哪个位置。 n是a序列的长度大小。

#include<bits/stdc++.h>
using namespace std;
int a[100001]={0};
int main()
{
	int n,m;
	int x;
	
	while(~scanf("%d %d",&n,&m))
	{
		for(int i=0;i<n;i++)
		scanf("%d",&a[i]);
		sort(a,a+n);

		while(m--)
		{
			int temp;
			scanf("%d",&temp);
			int p=upper_bound(a,a+n,temp)-a;
			printf("%d\n",p);
		}
	}
} 

附上一个学习链接：lower_bound和upper_bound的用法（C++）_理想不是梦33的博客-CSDN博客_lowerbound和upperbound