题目要求
蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问整数 x 是否在数组 A 中。
输入格式
第一行输入两个整数 n 和 m,分别表示数组的长度和查询的次数。
接下来一行有 n 个整数 ai。
接下来 m 行,每行有 1 个整数 x,表示蒜头君询问的整数。
输出格式
对于每次查询,如果可以找到,输出"YES"
,否则输出"NO"
。
数据范围
1≤n,m≤100000,0≤x≤1000000。
测试的数据可能会很大,要尽可能的降低时间复杂度来避免超时。因此排序选择快速排序,查找则使用二分查找。
#include<stdio.h>
void P(int a[],int x,int y);
int C(int n,int a[],int x);
int main(void){
int n,m;
scanf("%d %d",&n,&m);
int a[n],x;
int i;
for(i=0;i<n;i++){
scanf("%d",&a[i]);
}
int l=0,r=n-1;
P(a,l,r); //调用排序函数
while(m--){
scanf("%d",&x);
if(C(n,a,x)){ //判断查找结果
printf("YES\n");
}else{
printf("NO\n");
}
}
return 0;
}
int C(int n,int a[],int x){ //二分查找
int l=0;
int r=n-1;
int h=0;
while(l<=r){
h=(r+l)/2;
if(a[h]<x){
l=h+1;
}else if(a[h]>x){
r=h-1;
}else if(a[h]==x){
return 1; //返回查找结果
}
}
return 0;
}
void P(int a[],int x,int y){ //快速排序
if(x<y){
int l=x,r=y,z=a[l];
while(l<r){
while(l<r&&a[r]>=z){
r--;
}
if(l<r&&a[r]<z){
a[l++]=a[r];
}
while(l<r&&a[l]<=z){
l++;
}
if(l<r&&a[l]>z){
a[r--]=a[l];
}
}
a[l]=z;
P(a,x,l-1);
P(a,l+1,y);
}
}