二分思想(稍简单)
蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问在数组 A 中,大于等于 x 的最小值是多大?
输入格式
第一行输入两个整数 n和 m,分别表示数组的长度和查询的次数。
接下来一行有 n 个整数 a_i
i
。
接下来 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
思路
运用快排将输入的数组从大到小进行排列,再利用二分法进行比较,同时求出大于输入值的最小数,最后进行输出。
#include<stdio.h>
void quickSort(int left,int right,int str1[]){//快排
if(left>=right){
return ;
}
int key=str1[left];
int low=left;
int high=right;
int x,y;
while(low<high){
while(str1[high]>=key&&low<high){
high--;
}
while(low<high&&str1[low]<=key){
low++;
}
if(low<high){
int temp=str1[low];
str1[low]=str1[high];
str1[high]=temp;
}
}
str1[left]=str1[low];
str1[low]=key;
x=low-1;
y=low+1;
quickSort(left,x,str1);
quickSort(y,right,str1);
}int main(){
int n,m,i=0,num=0,num1=0,x=0,temp;
scanf("%d%d",&n,&m);
int str1[n]; //定义输入的数组
int str2[m]; //定义需寻找的数组
while(1){
scanf("%d",&num); //输入数组1
char c=getchar();
str1[i++]=num;
if(c=='\n'){
break;
}
}
for(i=0;i<m;i++){ //输入数组2
scanf("%d",&num1);
str2[i]=num1;
}
x=n-1;
quickSort(0,x,str1);
int left,right,mid,j=0,s=0;
for(i=0,j=0;i<m;i++){ //二分法进行比较
left=0;
right=n-1;
while(left<=right){
mid=(left+right)/2;
if(str1[mid]<str2[i]){
left=mid+1;
}
else if(str1[mid]>str2[i]){
right=mid-1;
}
else if(str1[mid]==str2[i]){
j=1;
printf("%d\n",str2[i]);//输出相等的数
break;
}
}
if(j!=1){
if(str2[i]>str1[n-1]){
printf("-1\n"); /输出应用-1表示大于数组1最大值的数
}
else {
printf("%d\n",str1[left]);//输出大于寻找数最小的数
}
}
else if(j==1){
j=0;//将判断数归零
}
}
return 0;
}