题目描述:
蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问整数 x 是否在数组 A 中。
输入格式:
第一行输入两个整数 n 和 m,分别表示数组的长度和查询的次数。
接下来一行有 n 个整数 ai。
接下来 m 行,每行有 1 个整数 x,表示蒜头君询问的整数。
输出格式:
对于每次查询,如果可以找到,输出"YES"
,否则输出"NO"
。
数据范围:
1≤n,m≤10^5,0≤x≤10^6。
输入样例:
10 5 1 1 1 2 3 5 5 7 8 9 0 1 4 9 10
输出样例:
NO YES NO YES NO
解题思路:完成基本输入后,对数组进行排序,采用时间复杂度较低的快速排序由小到大排序,之后使用二分法查询数。快速排序和二分法的思想是重点!代码段的注释里这两种思想体现的淋漓尽致!
#include<stdio.h>
void quicksort(int left,int right,int a[]);//定义一个快速排序的函数
int main()
{
int n, m;
scanf("%d %d", &n, &m);//输入数组的长度和查询的次数
int a[n];//定义数组
int i;
for (i = 0; i < n; i++) {//对数组初始化
scanf("%d", &a[i]);
}
quicksort(0,n-1,a);//对数组进行排序
int low = 0, high = n, cnt;
int mid, key,flag;
int s=1;
while (s<=m) {//查询的次数要小于等于m
cnt = 0;
flag=0;
low = 0;
high = n;
scanf("%d", &key);//输入要查询的数
while (low <= high && cnt <= m) {
//二分法:一开始让要查询的数与数组的中间项比较, 如果大于中间项,则把下限改为中间+1项,如果大于中间项,则把上限改为中间-1项,再与后面的中间项比较,逐次与新的中间项比较
mid = (low + high) / 2;
if (key > a[mid]) {//让查询的数与中间项对比,如果大于中间项,则把下限改为中间+1项
low = mid + 1;
}
if (key < a[mid]) {//如果大于中间项,则把上限改为中间-1项
high = mid - 1;
}
if (a[mid] == key) {//如果查询的数刚好为中间值,则找到这个数了
printf("YES\n");
flag=1;//把flag设为1 代表找到数了
break;
}
cnt++;//有查询次数限制,如果限制次数内找不到,则也默认查询数不在数组里
}
if(flag==0){//如果没找到数,则flag为0,就输出no
printf("NO\n");
}
s++;
}
return 0;
}
void quicksort(int left,int right,int a[])
{
if(left>right){//函数退出条件:要检测的下限下标大于上线下标
return ;
}
int temp=a[left];//以a[0]作为基准数
int i=left,j=right,t;
while(i<j){ //把比基准数小的全移到中间值左边,比基准数大的全移到中间值右边
while(a[j]>=temp&&j>i){//从第0项开始逐次递增,找到数组中比a[0]小的一项
j--;
}
while(a[i]<=temp&&j>i){//从第n项开始逐次递减,找到数组中比a[0]大的一项
i++;
}
if(j>i){//将比a[0]大的和比a[0]小的交换
t=a[i];
a[i]=a[j];
a[j]=t;
}
}
a[left]=a[i];//将中间值与a[0]
a[i]=temp;
quicksort(left,i-1,a);//递归思想:每次找到中间值后,把从0到中间值-1的项组成新数组进行排序
quicksort(i+1,right,a);//递归思想:每次找到中间值后,把从中间值+1到n的项组成新数组进行排序
return ;
}