题目描述
有一个数据字典,里面存有n个数字(n<=100000),小明现在接到一个任务,这项任务看起来非常简单——给定m个数字,分别查询这m个数字是否出现在字典之中;但是考虑到数据量的问题,小明找到了善于编程的你,希望你可以帮他解决这个问题。
输入
输入数据只有一组!
第一行包含两个整数n m,分别代表字典中数字的个数和要查询的数字的个数。
接着n行代表字典中的n个数字。
最后m表示要查询的数字。
输出
如果某个数字存在,则输出YES,否则输出NO
示例输入
5 3 1 2 3 4 5 5 4 10
示例输出
YES YES NO
#include<iostream> #include<string.h> #include<stdio.h> #include<ctype.h> #include<algorithm> #include<stack> #include<queue> #include<set> #include<math.h> #include<vector> #include<deque> #include<list> using namespace std; int a[100000]; int main() { int n,i,j,m,k,t; scanf("%d%d",&n,&m); memset(a,0,sizeof(a));//清零数组 for(i=0;i<n;i++) { scanf("%d",&k); a[k]=1;//把该数对应的数组A中的值赋为1 } while(m--) { scanf("%d",&t); if(a[t]==1)//如果该书对应的数组A中值为1,说明数组中有与该数相等的数 printf("YES\n"); else printf("NO\n"); } return 0; }
方法2:二分法
#include<stdio.h> #include<algorithm> using namespace std; int s[100010]; int main() { int m,n,i,j,key,ok,x,y,mid; scanf("%d%d",&m,&n); for(i=0;i<m;i++) scanf("%d",&s[i]); sort(s,s+m); for(j=0;j<n;j++) { scanf("%d",&key); ok=0; x=0;y=m-1; while(x<=y) { mid=(x+y)/2; if(key==s[mid]) { ok=1;break; } else if(key<s[mid]) y=mid-1; else x=mid+1; } if(ok) printf("YES\n"); else printf("NO\n"); } return 0; }