得到前k个小的数用优先队列，实际上也是用的堆排序，维持一个k大小的大顶堆，用make_heap相关几个函数就好了，直接sort的话如果数组很大白排了那么多的元素。

要是是前k大的数的话要多加个仿函数，自定义比较大小。

楼主你有问你面的部门做机器学习吗？我也是C++岗，不会机器学习好慌...