面试题40：最小的k个数(基于最大堆)

最新推荐文章于 2025-09-06 18:08:40 发布

原创最新推荐文章于 2025-09-06 18:08:40 发布 · 470 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种利用C++ STL中的最大堆（make_heap, push_heap, pop_heap, sort_heap）来解决求解前K小元素的问题。通过读取一系列整数并维护一个大小为K的最大堆，可以高效地找到前K小的数值。该方法适用于在线算法，能实时更新前K小元素。

//牛客网代码：https://www.nowcoder.com/questionTerminal/6a296eb82cf844ca8539b57c23e6e9bf
//make_heap讲解：https://blog.youkuaiyun.com/liu_sheng_1991/article/details/52298887

#include<iostream>
#include<vector>
//#include<functional>
#include<algorithm>

using namespace std;

int main()
{
   int num;
   vector<int> nums;
   int k;
   cin >> k;

while (cin >> num)
nums.push_back(num);

if (nums.size() < 0 || k<0 || k>nums.size())
return false;

vector<int> res(nums.begin(),nums.begin()+k);
make_heap(res.begin(), res.end()); //默认是最大堆

   for (int i = k; i < nums.size(); i++)
   {
       if (nums[i] < res[0])
       {
           pop_heap(res.begin(),res.end());
           res.pop_back();

res.push_back(nums[i]);
push_heap(res.begin(),res.end());

}

   }

   sort_heap(res.begin(),res.end());

for (int i = 0; i < res.size(); i++)
cout << res[i] << " ";

system("pause");

}

//3 2 6 8 1 2 9 45 2 3 9