题目描述
现在有一个长度为n的有序的序列,m次询问,每次会告诉你一个数字x,让你完成以下两种操作:
1.找到第一个大于等于x的值的下标,如果有多个等于的话找到第一个等于的下标
2.找到第一个大于x的值的下标 如果没有就输出NO,保证序列有序。
输入格式
第一行两个正整数n(n <= 100000)代表序列长度和m(m <= 100000)代表询问次数,
第二行n个数字表示有序序列a(其中0 <= ai <= 200000)。接下来m行每行一个数字x(0 <= x <= 200000)。
输出格式
答案输出m行,每行两个数字分别表示答案,两个答案之间用空格隔开。
输入输出样例
输入
5 5
1 3 4 5 6
1
2
3
6
7
输出
1 2
2 2
2 3
5 NO
NO NO
首先尝试直接暴力破解
#include<iostream>
using namespace std;
int main()
{
int n,m;
cin >> n >> m ;
int a[n+1];
for(int i=1;i<=n;i++)
{
cin >> a[i];
}
while(m--)
{
int x;
cin >> x ;
for(int i=1;i<=n;i++)
{
if(a[i]>=x){
cout << i << " " ;
break;
}
else if(i==n) cout << "NO" << " " ;
}
for(int i=1;i<=n;i++)
{
if(a[i]>x){
cout << i << endl ;
break;
}
else if(i==n) cout << "NO" << endl ;
}
}
}
当输入规模过大,时间会超限,考虑用二分查找算法
#include<iostream>
using namespace std;
void cuthalfSearch1(int array[], int n, int target) {
int low = 0, high = n-1, middle = 0,res=high;
while(low <= high) {
middle = (high+low)/2;
if(array[middle] >= target) {
res=middle;
high = middle-1;
}
else
{
low = middle + 1;
}
}
if(array[res] >= target) cout << res+1 << " " ;
else cout << "NO" << " " ;
}
void cuthalfSearch2(int array[], int n, int target) {
int low = 0, high = n-1, middle = 0,res=high;
while(low <= high) {
middle = (high+low)/2;
if(array[middle] > target) {
res=middle;
high = middle-1;
}
else
{
low = middle + 1;
}
}
if(array[res] > target) cout << res+1 << endl ;
else cout << "NO" << endl ;
}
int main()
{
int n,m;
cin >> n >> m ;
int a[n];
for(int i=0;i<n;i++)
{
cin >> a[i];
}
while(m--)
{
int x;
cin >> x ;
cuthalfSearch1(a,n,x);
cuthalfSearch2(a,n,x);
}
return 0;
}