蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问整数 x 是否在数组 A 中。
输入格式
第一行输入两个整数 n 和 m,分别表示数组的长度和查询的次数。
接下来一行有 n 个整数 ai。
接下来 m 行,每行有 1 个整数 x,表示蒜头君询问的整数。
输出格式
对于每次查询,如果可以找到,输出"YES",否则输出"NO"。
数据范围
1≤n,m≤105,0≤x≤106。
Sample Input
10 5
1 1 1 2 3 5 5 7 8 9
0
1
4
9
10
Sample Output
NO
YES
NO
YES
NO
解题思路
1.先排序使函数变有序
2.用二分法查找有序数组
排序基本代码
void find(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;
find(s,l,i-1);
find(s,i+1,r);
}
二分法基本代码
int search(int n,int num[],int x){ //函数二分法
int i=0,j=n-1,k=0;
while(i<=j){
k=(j+i)/2;
if(x>num[k]){
i=k+1;
}else if(x<num[k]){
j=k-1;
}else if(num[k]==x){
return 1;
}
}
return 0;
}
最终代码
#include<stdio.h>
void find(int s[], int l,int r);
int search(int n,int num[],int x);
int main(void){
int n,m,t;
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;
find(a,i,t); //调用函数快速排序法
for(j=0;j<m;j++){
scanf("%d",&b);
if(search(n,a,b)){ //二分法查找
printf("YES\n");
}else{
printf("NO\n");
}
}
return 0;
}
int search(int n,int num[],int x){ //函数二分法
int i=0,j=n-1,k=0;
while(i<=j){
k=(j+i)/2;
if(x>num[k]){
i=k+1;
}else if(x<num[k]){
j=k-1;
}else if(num[k]==x){
return 1;
}
}
return 0;
}
void find(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;
find(s,l,i-1);
find(s,i+1,r);
}
}