蒜头君手上有个长度为 nn 的数组 AA。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问整数 xx 是否在数组 AA 中。
输入格式
第一行输入两个整数 nn 和 mm,分别表示数组的长度和查询的次数。
接下来一行有 nn 个整数 a_iai。
接下来 mm 行,每行有 11 个整数 xx,表示蒜头君询问的整数。
输出格式
对于每次查询,如果可以找到,输出"YES"
,否则输出"NO"
。
数据范围
1 \le n, m \le 10^5, 0 \le x \le 10^61≤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
解题思路
先用快速排序法进行排序,之后用二分法查找数据
#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;
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(search(n,a,b)){ //二分法查找
printf("YES\n");
}else{
printf("NO\n");
}
}
return 0;
}
int search(int n,int num[],int x){ //函数二分法
int low=0,high=n-1,mid=0;
while(low<=high){
mid=(high+low)/2;
if(x>num[mid]){
low=mid+1;
}else if(x<num[mid]){
high=mid-1;
}else if(num[mid]==x){
return 1;
}
}
return 0;
}
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);
}
}