题目:n个数的数组,找到最小的k个数字
分析:可以使用排序然后取前K个数字,也可以维护一个k大小的set,set的存储是有序的,当set不满k个时候就入set,如果set达到k就与set中最大值比较,如果比最大值还大,那么继续,如果比最大值小那么替换最大值。set是用红黑树实现的。
void FindMinK(int arr[],int length,int k)
{
set<int> mink;
for (int i=0; i<length; i++) {
if (mink.size()<k) {
mink.insert(arr[i]);
}
else{
int temp=*mink.rbegin(); //获得最大值
if (arr[i]>temp) {
continue;
}
else{
mink.erase(temp);
mink.insert(arr[i]);
}
}
}
for (set<int>::iterator b=mink.begin();b!=mink.end();b++) {//!=
printf("%d \n",*b);//*b
}
}
这里说一下set的用法
set默认是从小到大的,如何获得最大的元素,rbegin()返回反向指针
*p就是获得迭代器中得值
erase(int key)删除所有关键字是key的
set<int>::iterator b 迭代器的写法