题目描述:蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问在数组 A 中,大于等于 x 的最小值是多大?
输入格式:
第一行输入两个整数 n 和 m,分别表示数组的长度和查询的次数。
接下来一行有 n个整数 ai。
接下来 m 行,每行有 1 个整数 x,表示蒜头君询问的整数。
输出格式
对于每次查询,如果可以找到,输出这个整数。
否则输出 -1。
数据范围
1 ≤ n, m ≤ 10^5,0≤x≤10^6。
思路:输入所需数据,用函数进行数组排序,之后用二分法进行分离判断,根据题目要求进行查询输出
(1)定义参数,数组
int n,m,t,o;
int i,k=0,j;
scanf("%d %d",&n,&m);
int a[n],b;
for(i=0;i<n;i++){
scanf("%d",&a[i]);
}
(2)进行边界检测,并用函数进行查找
i=0;
t=n-1;
quick(a,i,t);
for(j=0;j<m;j++){
scanf("%d",&b);
if(b>a[n-1]){
printf("-1\n");
}else if(b<=a[0]){
printf("%d\n",a[0]);
}else{
o=search(n,a,b);
printf("%d\n",o);
}
}
return 0;
(3)判断条件是否符合,返回值
int search(int n,int num[],int x){ //查找大于等于的第一个数
int low=0,high=n-1,mid=0;
while(low<=high){
mid=low+(high-low)/2;
if(x>num[mid]){
low=mid+1;
}else if(x<=num[mid]){ //小于等于数
high=mid-1;
}
}
return num[low]; //返回最小值
}
(4)进行排序
void quick(int s[], int l,int r){
if (l<r){
int i=l,j=r,x=s[l];
while (i<j){
while(i<j &&s[j]>=x){
j--;
}
if(i<j&&s[j]<x){
s[i++]=s[j];
}
while(i<j&&s[i]<=x){
i++;
}
if(i<j&&s[i]>x){
s[j--]=s[i];
}
}
s[i]=x;
quick(s,l,i-1);
quick(s,i+1,r);
}
}
完整代码:
#include<stdio.h>
void quick(int s[], int l,int r);
int search(int n,int num[],int x);
int main(void){
int n,m,t,o;
int i,k=0,j;
scanf("%d %d",&n,&m);
int a[n],b;
for(i=0;i<n;i++){
scanf("%d",&a[i]);
}
i=0;
t=n-1;
quick(a,i,t);
for(j=0;j<m;j++){
scanf("%d",&b);
if(b>a[n-1]){
printf("-1\n");
}else if(b<=a[0]){
printf("%d\n",a[0]);
}else{
o=search(n,a,b);
printf("%d\n",o);
}
}
return 0;
}
int search(int n,int num[],int x){
int low=0,high=n-1,mid=0;
while(low<=high){
mid=low+(high-low)/2;
if(x>num[mid]){
low=mid+1;
}else if(x<=num[mid]){
high=mid-1;
}
}
return num[low];
}
void quick(int s[], int l,int r){
if (l<r){
int i=l,j=r,x=s[l];
while (i<j){
while(i<j &&s[j]>=x){
j--;
}
if(i<j&&s[j]<x){
s[i++]=s[j];
}
while(i<j&&s[i]<=x){
i++;
}
if(i<j&&s[i]>x){
s[j--]=s[i];
}
}
s[i]=x;
quick(s,l,i-1);
quick(s,i+1,r);
}
}