有问题欢迎评论我们互相学习!
堆的概念及结构
现实中我们通常把堆(一种二叉树)使用顺序结构的数组来存储,需要注意的是这里的堆和操作系统虚拟进程地址空间中的堆是两回事,一个是数据结构,一个是操作系统中管理内存的一块区域分段。
堆的性质:
&&堆中某个节点的值总是不大于或不小于其父节点的值;
&&堆总是一棵完全二叉树。
小根堆:父节点总是比子节点小。
大根堆:父节点总是比子节点大。
堆排序
#include<iostream>
#include<vector>
using namespace std;
//向下调整 大堆(如果想建立小堆就把比较符号改反就可以了)
void AdjustDown(vector<int> &a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
//选择子节点最大的与父节点比较
if (child + 1 < n && a[child + 1] > a[child])
{
child = child + 1;
}
//子节点如果比父亲大二者交换并更新父节点位置继续向下调整
if (a[child] > a[parent])
{
swap(a[child], a[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
//已经有序了无需排序,退出
break;
}
}
}
//堆排序 排升序建大堆 降序建小堆
void HeapSort(vector<int> a)
{
int n = a.size();
//1.建大堆O(N)
//n是size 最后一个元素是 size - 1 = n - 1 parent = (最后一个元素-1)/2
for (int i = (n-1-1)/2 ; i >= 0; i--)
{
AdjustDown(a, n, i);
}
cout << "建立大堆后" << endl;
for (auto e : a)
{
cout << e << " ";
}
cout << endl;
//依次调堆O(N*logN)
//2.把大堆堆顶与堆尾互换,然后对从堆顶元素 向队尾-- 的元素AdjustDown
for (int i = n - 1; i > 0; i--)
{
swap(a[0], a[i]);
AdjustDown(a, i, 0);
}
cout << "调整为升序" << endl;
for (auto e : a)
{
cout << e << " ";
}
cout << endl;
}
int main()
{
vector<int>a;
a.push_back(26);
a.push_back(2);
a.push_back(5);
a.push_back(3);
a.push_back(9);
a.push_back(4);
a.push_back(8);
a.push_back(5);
a.push_back(1);
a.push_back(7);
for (auto e : a)
{
cout << e <<" ";
}
cout << endl;
HeapSort(a);
system("pause");
return 0;
}
这里解释下为什么排升序要建立大堆,降序建立小堆
TopK问题
TOP-K问题:即求数据结合中前K个最大的元素或者最小的元素,一般情况下数据量都比较大。
比如:专业前10名、世界500强、富豪榜、游戏中前100的活跃玩家等。
对于Top-K问题,能想到的最简单直接的方式就是排序,但是:如果数据量非常大,排序就不太可取了(可能
数据都不能一下子全部加载到内存中)。最佳的方式就是用堆来解决,基本思路如下:
- 用数据集合中前K个元素来建堆
前k个最大的元素,则建小堆
前k个最小的元素,则建大堆 - 用剩余的N-K个元素依次与堆顶元素来比较,不满足则替换堆顶元素
- 将剩余N-K个元素依次与堆顶元素比完之后,堆中剩余的K个元素就是所求的前K个最小或者最大的元素。
#include<iostream>
#include<vector>
using namespace std;
//向下调整 大堆
void AdjustDown(vector<int> &a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
//选择子节点最大的与父节点比较
if (child + 1 < n && a[child + 1] > a[child])
{
child = child + 1;
}
//子节点如果比父亲大二者交换并更新父节点位置继续向下调整
if (a[child] > a[parent])
{
swap(a[child], a[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
//已经有序了无需排序,退出
break;
}
}
}
//取一大堆数据的前k个最小值
void TopK(vector<int> &a, int k)
{
int n = a.size();
vector<int>v;
for (int i = 0; i < k ; i++)
{
v.push_back(a[i]);
}
//建立大堆
for (int i = (k - 1 - 1) / 2; i >= 0; i--)
{
AdjustDown(v, k, i);
}
cout << endl;
//剩下的N-K个数跟堆顶的数据比较,比他小,就替换他进堆
for (int i = k; i < n; i++)
{
if (a[i] < v[0])
{
v[0] = a[i];
AdjustDown(v, k, 0);
}
}
for (auto e : v)
{
cout << e << " ";
}
cout << endl;
}
int main()
{
vector<int>a;
a.push_back(1);
a.push_back(26);
a.push_back(2);
a.push_back(3);
a.push_back(9);
a.push_back(4);
a.push_back(8);
a.push_back(5);
a.push_back(1);
a.push_back(7);
for (auto e : a)
{
cout << e <<" ";
}
cout << endl;
TopK(a, 3);
system("pause");
return 0;
}