蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问在数组 A 中,大于等于 x 的最小值是多大?
输入格式
第一行输入两个整数 n 和 m,分别表示数组的长度和查询的次数。
接下来一行有 n 个整数 ai。
接下来 m 行,每行有 1 个整数 x,表示蒜头君询问的整数。
输出格式
对于每次查询,如果可以找到,输出这个整数。
否则输出 −1。
数据范围
1≤n,m≤10^5,0≤x≤10^6。
Sample Input
10 5 1 1 1 2 3 5 5 7 8 9 0 1 4 9 10
Sample Output
1 1 5 9 -1
这是一道数据量很大的题目,在一个数据量极大的数组中数字,按理来说枚举发应该是最简单易懂的了,但是所耗费的时间一定也很大,那么我们应该怎么办。
这里我们使用二分法。
那么二分法是啥,可以参考我之前写二分法教程。
要用二分法就一定要排序,这里建议使用快速排序,快排对于数据量大的数组还是很有效的。
void quicksort(int left,int right,int str[])
{
int i,j,temp,t;
if(left>right)
return;
temp=str[left];
i=left;
j=right;
while(i!=j)
{
while(str[j]>=temp&&j>i)
{
j--;
}
while(str[i]<=temp&&j>i)
{
i++;
}
if(i<j)
{
t=str[i];
str[i]=str[j];
str[j]=t;
}
}
str[left]=str[i];
str[i]=temp;
quicksort(left,i-1,str);
quicksort(i+1,right,str);
return;
}
以上是快排基本思想。
但这道题并不是让我们找我们输入的数字,而是去寻找大于等于 x 的最小值是多大。
这可咋整
那么这里需要提到两个值得注意的的点:
1、若是输入的数小于数组的最小数,则可以直接输出数组的最小数。
2、若是输入的数大于数的最大数,则可以直接输出-1。
if(j>str[n]) //大于最大数
printf("-1\n");
else if(j<=str[1])
printf("%d\n",str[1]); //小于最小数
最重要的一步便是寻找那个大于等于x的最小值了,这依然还是有两种情况:
1、如果x等于数组中的任意一个数,那么输出那个数。
2、如果x不等于数组中的任意一个数,则需要依靠代码去实现
int Bin_Search(int *str,int cnt, int target)
{
int first=1,last=cnt,mid;
while(first<=last)
{
mid=(first+last)/2;
if(str[mid]<target)
first=mid+1;
else if(str[mid]>=target)
last=mid-1;
}
return str[first];
}
所以总代码如下:
#include <stdio.h>
void quicksort(int left,int right,int str[]) //快排
{
int i,j,temp,t;
if(left>right)
return;
temp=str[left];
i=left;
j=right;
while(i!=j)
{
while(str[j]>=temp&&j>i)
j--;
while(str[i]<=temp&&j>i)
i++;
if(i<j)
{
t=str[i];
str[i]=str[j];
str[j]=t;
}
}
str[left]=str[i];
str[i]=temp;
quicksort(left,i-1,str);
quicksort(i+1,right,str);
return;
}
int Bin_Search(int *str,int cnt, int target) //二分查找
{
int first=1,last=cnt,mid;
while(first<=last)
{
mid=(first+last)/2;
if(str[mid]<target)
first=mid+1;
else if(str[mid]>=target)
last=mid-1;
}
return str[first];
}
int main(void)
{
int n,m,i,j,o;
scanf("%d%d",&n,&m);
int str[n];
for(i=1;i<=n;i++)
scanf("%d",&str[i]);
quicksort(1,n,str);
while(m--) //输入m次
{
scanf("%d",&j);
if(j>str[n])
printf("-1\n");
else if(j<=str[1])
printf("%d\n",str[1]);
else
{
o=Bin_Search(str,n,j);
printf("%d\n",o);
}
}
return 0;
}