AcWing 789. 数的范围

 

给定一个按照升序排列的长度为 n𝑛 的整数数组，以及 q𝑞 个查询。

对于每个查询，返回一个元素 k𝑘 的起始位置和终止位置（位置从 00 开始计数）。

如果数组中不存在该元素，则返回 -1 -1。

输入格式

第一行包含整数 n𝑛 和 q𝑞，表示数组长度和询问个数。

第二行包含 n𝑛 个整数（均在 1∼100001∼10000 范围内），表示完整数组。

接下来 q𝑞 行，每行包含一个整数 k𝑘，表示一个询问元素。

输出格式

共 q𝑞 行，每行包含两个整数，表示所求元素的起始位置和终止位置。

如果数组中不存在该元素，则返回 -1 -1。

数据范围

1≤n≤1000001≤𝑛≤100000
1≤q≤100001≤𝑞≤10000
1≤k≤100001≤𝑘≤10000

输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

先写个暴力：

#include<bits/stdc++.h>

using namespace std;

const int N=1e5+10;
int n,q;
int a[N],res[N],q1[10005];
//归并排序一下，（虽然题目已经说了是升序）（绝对不是我没有认真读题QWQ）
void merge_quick(int a[],int l,int r)
{
	if(l>=r) return;
	int mid= l+r >> 1;
	
	merge_quick(a,l,mid),merge_quick(a,mid+1,r);
	
	int i=l,j=mid+1,k=0;;
	
	while(i<=mid&&j<=r)
	{
		if(a[i]>=a[j]) res[k++]=a[j++];
		else res[k++]=a[i++];
	}
	
	while(i<=mid) res[k++]=a[i++];
	while(j<=r) res[k++]=a[j++];
	
	for(int i=l,k=0;i<=r;i++,k++)
	a[i]=res[k];	
}
int main()
{
	cin >> n >> q;
//标志变量flag，判断该数组中是否含有所要查询的数
	int flag=1;
	for(int i=0;i<n;i++) cin >> a[i];
	merge_quick(a,0,n-1);
	for(int i=0;i<q;i++)
	{
	    cin >> q1[i];
	    for(int j=0;j<n;j++)
	    {
	    	if(a[j]==q1[i])
	    	flag=0;
		}
//格式小细节
		if(flag==1&&i!=q-1) cout << "-1 -1" << endl;
		if(flag==1&&i==q-1) cout << "-1 -1";
		flag=1;
		for(int m=0;m<n;m++)
		{
//判断所在位置并输出
			if(a[m]==q1[i]&&m==0)
//如果第一个数就等于所要寻找的数，那么输出第一个数的左边界
			cout << "0 " ;
如果前一个数不等于，后一个数等于，那么即找到左边界
			if(a[m]!=q1[i]&&a[m+1]==q1[i])
			cout << m+1 << " ";
//右边界/注意格式
			if(a[m]==q1[i]&&a[m+1]!=q1[i]&&i!=q-1)
			cout << m << endl;
			if(a[m]==q1[i]&&a[m+1]!=q1[i]&&i==q-1)
			cout << m;
		}
	}
	return 0;
}

暴力肯定拿不满(⊙﹏⊙)

既然是查询位置，而且题目给的是升序（若不是，可以用sort函数排列一下），所以想到二分法求解，利用二分的性质，左右边界可以二分在不同的区间求解：

代码的具体细节在下一篇

#include<bits/stdc++.h>

using namespace std;

const int N=1e5+10;

int n,q,a[N];

int main()
{
    cin >> n >> q;
    for(int i=0;i<n;i++) cin >> a[i];
    
    while(q--)
    {
    	int m;
		cin >> m;
    	int l=0,r=n-1;
//左边界
    	while(l<r)
    	{
    		int mid= l+r >>1;
    		if(a[mid]>=m) r=mid;
    		else l=mid+1;
		}
		if(a[l]!=m) cout << "-1 -1"<< endl;
//右边界
		else
		{
			cout << l << " ";
			l=0,r=n-1;
			while(l<r)
			{
				int mid= l+r+1 >>1; 
				if(a[mid]<=m) l=mid;
				else r=mid-1;
			}
			cout << l << endl;
		}
	}
	return 0;
}