您的位置:首页 > 编程语言 > Go语言

Algorithm中Heap

2016-07-29 17:29 295 查看
介绍algorithm头文件中根的相关函数

make_heap(首位置, 尾位置+1, 可选的cmp函数); -> 构造堆

push_heap(首位置, 尾位置+1, 可选的cmp函数); -> 添加元素到堆中

pop_heap(首位置, 尾位置+1, 可选的cmp函数);  -> 从堆中移出元素

sort_heap(首位置, 尾位置+1, 可选的cmp函数); -> 整个堆进行排序

#include<algorithm>
#include<cstdio>
using namespace std;

bool inc_cmp(int a,int b){ return a > b; }
// 可以理解为:左a右b,当a>b的时候,双方交换位置,
// 所以上式这个cmp意为从小到大(排序)。

bool des_cmp(int a, int b){ return a < b; }

int num[10]={3, 1, 2, 5, 6, 4};

int main()
{
make_heap(num, num+6);//构造一个堆,默认是大根堆。
// 上式等价于make_heap(&num[0], &num[6]);可以看到,区间是[0, 7),
// 即参数中的末尾是真真实末尾的后一个位置。
// 可以修改为make_heap(num, num+6, inc_cmp); 此时为小根堆。

// 插入一个元素到先前创建的堆中,插进来的数将调整到合适的位置。
num[6] = 5;
push_heap(num, num+7);

// pop_heap将堆顶数据移动到末尾位置,然后将剩余数据重新构造堆。
// 其中一个易于理解的例子就是,不断将大根堆堆顶移到末尾,
// 以此模拟从小到大排序。

/* ---------- */
printf("before: ");
for(int i = 0; i < 7; i++)
printf("%d ", num[i]);
printf("\n");

for(int i = 7; i >= 1; i--) // 共7个数,交换到最后第二个数为止。
{
pop_heap(num, num+i);
}

printf("after : ");
for(int i = 0; i < 7; i++)
printf("%d ", num[i]);
printf("\n");
/* ---------- */

//sort_heap,堆排序,因为前面都pop_heap掉了,所以得重新make_heap
make_heap(num, num + 6);
sort_heap(num, num+6, des_cmp);
for(int i = 0; i < 7; i++)
printf("%d ", num[i]);
printf("\n");
return 0;
}
总结:
想要进行堆操作,前提必须是“已构建了堆”。
如果在使用时碰到数据混乱,那么很有可能就是没有make_heap。
需要注意的是make_heap后进行pop_heap操作,那么这个根比原来的"影响范围"就小了,
假设进行上述的"排序"操作,那么这些数据就不再是堆。
想要另行操作只能重新make_heap。 
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  heap