剑指off-找到最小k个数字

来源：互联网发布：股东大会网络投票平台编辑：程序博客网时间：2024/05/16 23:45

题目：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 迭代器的写法

0 0