排序与检索
E.g. 现有N个大理石,每个大理石上写了一个非负整数。首先把每个数从小到大排序,然后回答Q个问题。每个问题问是否有一个大理石写着某个整数x。如果是,要回答在哪个大理石上写着x。排序后的大理石从左到右编号为1~N。
solution: 先排序,后查找
关于排序
sort使用数组元素默认的大小比较运算符进行排序,可以对任意对象进行排序,不一定是内置类型,如果希望用sort排序,这个类型需要定义“小于”运算符,或者是在排序时传入一个“小于”函数。
int a[n];
排序对象可以存在于普通数组里,sort(a,a+n);
也可以存在于vector中,sort(v.begin(),v.end());
用sort进行排序之后,可以使用low_bound(a, a+n, x),查找大于或等于x的第一个位置。
int p=low_bound(a, a+n, x)-a;//在已排序数组中寻找x
if(a[p]==x)
....
还有一个unique函数可以删除有序数组中的重复元素,对有序的容器重新排列,将第一次出现的元素从前往后排,其他重复出现的元素依次排在后面,返回迭代器,迭代器指向的是重复元素的首地址。
int a[14]={1,1,1,2,2,3,3,3,4,4,5,6,7,7}
int result=unique(a,a+14)-a;//result的返回值是7
sort 和 low_bound 的定义在头文件 #include<algorithm>
中
例题解法:
#include<cstdio>
#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 loc=low_bound(a,a+n,x)-a;
if(a[loc]==x)
print("%d found at %d\n",x,loc+1);
else printf("%d not found\n",x);
}
}
return 0;
}