Where is the Marble? UVA - 10474

Where is the Marble?

 UVA - 10474 

有n个石头，每个石头上都有数字，每组输入需要你找若干个数字，如果没找到就输出not find ，找到了就输出第一次见到这个数的位置。

看上去很简单，写起来也简单，然后第一发就过了。

#include <bits/stdc++.h> 
using namespace std;

int main(){
	int n,m;
	int a[10005];
	int num=1; 
	while(cin>>n>>m&&n&&m)
	{
		for(int i=1;i<=n;i++)
			cin>>a[i];
		printf("CASE# %d:\n",num++);
		sort(a+1,a+n+1);
		for(int j=1;j<=m;j++)
		{
			
			int num,flag;
			cin>>num;
		//	printf("%d\n",num);
			flag=0;
			for(int i=1;i<=n;)
			{
				if(num<a[i])
				{
					flag=-1;
					break;
				}
				else if(num==a[i])
				{
					flag=i;
					break;
				}
				else 
					i++;
			}
			
			if(flag<1)
				printf("%d not found\n",num);
			else
				printf("%d found at %d\n",num,flag);
		}
	}
	return 0;
}

虽然直接就过了，但是我的代码其实存在很多的问题，在查找题目要求的数字所用的是顺序搜索，时间复杂度是O(n),如果数据量稍大可能会超时，在之后我对代码进行了重写，使用了c++的stl库的函数，不仅减少了时间复杂度，还精简了代码

 

AC代码如下

#include <bits/stdc++.h> 
using namespace std;

int main(){
	int n,m;
	int a[10005];
	int num=1;
	while(cin>>n>>m&&n&&m)
	{
		
		for(int i=1;i<=n;i++)
			cin>>a[i];
		sort(a+1,a+n+1);
		printf("CASE# %d:\n",num++);
		for(int i=1;i<=m;i++)
		{
			int b;
			cin>>b;
			int p=lower_bound(a+1,a+n+1,b)-a;//库函数，在已排序的数组中找数字b 
			if(a[p]==b)
				printf("%d found at %d\n",b,p);
			else
				printf("%d not found\n",b);
		}
	}
	return 0;
}

在代码中我使用了lower_bound函数，这个函数的作用就是使用二分查找找到所需要查找的数的最低的地址，与函数相对的还有一个upper_bound,具体可以参考以下链接https://blog.csdn.net/zifengningyu/article/details/80159892。