题目描述
给定一个集合,查找元素是否在集合中出现。
输入
每个测试用例由多行组成,第一行是两个整数n和m,两个数范围在1到100000之间。自第二行起一共有n+m个整数,其中前面n个整数代表集合的元素,随后的m个整数是待查询的数。所有的整数在范围[-2^31,2^31)内。
输出
对于每个待查询的数,如果在集合中则输出yes,否则输出no.
样例输入
5 3
7 9 3 2 -5
4 9 -5
5 3
-2 1 0 -2 1
0 -2 3
样例输出
no
yes
yes
yes
yes
no
1普通的方法时间复杂度较大,很容易超限,所以思考之后用二分查找
2注意调用函数会耗一定的时间,
#include<stdio.h> #include<algorithm> using namespace std; int a[300000]; int n; int main() { int m,i,j,num,left,right,mid,flag; while(scanf("%d %d",&n,&m)!=EOF) { if(!n&&!m) break; for(i=1;i<=n;i++) scanf("%d",&a[i]); sort(a+1,a+1+n); for(i=0;i<m;i++) { scanf("%d",&num); left=1;right=n; flag=0;mid; while(left<=right) { mid=(right+left)/2; if(a[mid]>num) right=mid-1; else if(a[mid]<num) left=mid+1; else if(a[mid]==num) {flag=1;break;} } if(flag) printf("yes\n"); else printf("no\n"); } } return 0; }