怎样从vector中构建大顶堆或者小顶堆,网上多见以int型为例。
但现实中复杂数据比较多,多用结构体表示。
本文演示如何从结构体类型的vector构建堆。
///ttt.cpp
//c++编译器下通过
#include<stdio.h>
#include<stdlib.h>
#include<vector>
#include<algorithm>
#include<iostream>
using namespace std;
struct node
{//结构体类型
int data;//数据
int count;//出现的个数
};
bool cmp_big(node x,node y)
{//大顶堆的比较函数
return x.data<y.data;
}
bool cmp_small(node x,node y)
{//小顶堆的比较函数
return !cmp_big(x,y);
}
void print(vector<node> elem)
{
for(int i=0;i<elem.size();i++)
{
printf("%d %d\n",elem[i].data,elem[i].count);
}
}
int main()
{
node a[]={{1,1},{3,1},{5,1},{7,1},{9,1},{2,1},{4,1},{6,1},{8,1}};
vector<node> elem;
elem.insert(elem.begin(),a,a+9);
//输出小顶堆
make_heap(elem.begin(),elem.end(),cmp_big);
print(elem);
cout<<"\n";
//输出大顶堆
elem.clear();
elem.insert(elem.begin(),a,a+9);
make_heap(elem.begin(),elem.end(),cmp_small);
print(elem);
return 0;
}