STL--排序与检索

题目

现有N个大理石，每个大理石上写了一个非负整数。首先把各数从小到大排序，然后回答Q个问题。每个问题是否有一个大理石写着某个整数x,如果是，还要回答哪个大理石写着x。排序后的大理石从左到右编写为1-N。（样例中，所有大理石上的数合并到一行，所有问题也合并到一行）

输入输出样例

输入样例

4 1
2 3 5 1
5 
5 2
1 3 3 3 1
2 3 

输出样例

CASE# 1:
5 found at 4
CASE# 2:
2 not found
3 found at 3

代码

#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 10000;
int main(){
	int n,q,x,a[maxn],kase=0;
	while(scanf("%d%d",&n,&q)==2&&n){
		printf("CASE# %d:\n",++kase);
		for(int i=0;i<n;i++){
			scanf("%d",&a[i]);
		}
		sort(a,a+n);
		while(q--){
			scanf("%d",&x);
			int p=lower_bound(a,a+n,x)-a;
			if(a[p]==x){
				printf("%d found at %d\n",x,p+1);
			}
			else{
				printf("%d not found\n",x);
			}
		}
	}
	return 0;
} 

? ? ? ?上面代码中的sort函数可以对任意对象进行排序，不一定是内置类型。如果希望用sort排序，这个函数需要定义“小于”运算符，或者在排序时传入一个“小于”函数。排序对象可以存在于普通数组里，也可以存在于vector中。前者用sort(a,a+n)的方式调用，后者用sort(v.begin(),v.end())的方式调用。lower_bound的作用是“大于或者等于x的第一个位置”。