蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问整数 x 是否在数组 A中。
输入格式
第一行输入两个整数 n和 m,分别表示数组的长度和查询的次数。
接下来一行有 n 个整数
接下来 m 行,每行有 1 个整数 x,表示蒜头君询问的整数。
输出格式
对于每次查询,如果可以找到,输出"YES",否则输出"NO"。
数据范围
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
- NO
- YES
- NO
- YES
- NO
解题思路:
因为这道题目的运行时间有限制,需要用到快排,所以自定义函数做快排运算。根据排好的数组用二分法来找到key值。找到则输出即可
1. #include<stdio.h>
2. void qu(int s[], int l,int r);
3. int se(int n,int nu[],int x);
4. int main(void){
5. int n,m,t;
6. int i,k=0,j;
7. scanf("%d %d",&n,&m);
8. int a[n],b;
9. for(i=0;i<n;i++){
10. scanf("%d",&a[i]); }
11. i=0;
12. t=n-1;
13. qu(a,i,t); //调用函数快速排序法
14. for(j=0;j<m;j++){
15. scanf("%d",&b);
16. if(se(n,a,b)){ //二分法查找
17. printf("YES\n");
18. }else{
19. printf("NO\n");
20. }
21. }
22. return 0;
23. }
24. void qu(int s[], int l,int r){ //函数快速排序
25. if (l<r){
26. int i=l,j=r,x=s[l];
27. while (i<j){
28. while(i<j &&s[j]>=x){
29. j--;
30. }
31. if(i<j&&s[j]<x){
32. s[i++]=s[j];
33. }
34. while(i<j&&s[i]<=x){
35. i++;
36. }
37. if(i<j&&s[i]>x){
38. s[j--]=s[i];
39. }
40. }
41. s[i]=x;
42. qu(s,l,i-1);
43. qu(s,i+1,r);//这里用到了递归的思路
44. }
45. }
46. int se(int n,int nu[],int x){ //函数二分法
47. int low=0,high=n-1,mid=0;
48. while(low<=high){
49. mid=(high+low)/2;
50. if(x>nu[mid]){
51. low=mid+1;
52. }else if(x<nu[mid]){
53. high=mid-1; }
54. else if(nu[mid]==x){
55. return 1; }
56. }
57. return 0;
58. }