堆基本操作实现最大堆

复制代码 代码如下:

#include
#include
#include
#include
#include

using namespace std;
const int M = 10003;

//定义数据节点
class dNode{
public:
 string name;
 int age;
 double score;
 dNode():name("no name"), age(0), score(0.0){}
 dNode(string name, int age, double score):name(name), age(age), score(score){}
 bool operator < (const dNode & d){
  return score < d.score;
 }
 bool operator > (const dNode &d){
  return score > d.score;
 }
 bool operator = (const dNode &d){
  name = d.name;age=d.age;score=d.score;
 }
 bool operator == (const dNode &d){
  return name == d.name && age == d.age && score == d.score;
 }
 void swap(dNode & a, dNode & b){
  dNode tmp = a;
  a = b;
  b = tmp;
 }
 void show(){
  cout << "***************" << endl;
  cout << "name: " << name << endl;
  cout << "age: " << age << endl;
  cout << "score: " << score << endl;
 }
};
//定义堆
template
class Heap{
public:
 dNode h[M];
 void heapify(int cur);
 int n;
 //数组下标从0开始
 int L(int i){return (i << 1) + 1;}
 int R(int i){return (i << 1) + 2;}
 int P(int i){return (i - 1) >> 1;}
public:
 Heap():n(0){}
 Heap(T data[], int len){
  memcpy(h, data, sizeof(T) * len);
  n = len;
 }
 //对数组h建立成堆
 void build();
 //插入一个元素
 void insert(T data);
 //弹出堆顶元素
 void pop(){
  h[0] = h[--n];
  heapify(0);
 };
 //堆顶元素
 T top(){return h[0];}
 //打印数组中的全部元素
 void show(){
  for(int i = 0; i < n; i++){
   cout << "***************" << endl;
   cout << "cur: " << i << endl;
   cout << "name: " << h[i].name << endl;
   cout << "age: " << h[i].age << endl;
   cout << "score: " << h[i].score << endl;
  }
 }

};

template
void Heap::build(){
 for(int i = (n / 2) - 1; i >= 0; i--){
  heapify(i);
 }
}

template
void Heap::insert(T data){
 h[n++] = data;
 T tmp = data;  //将新增的节点保存
 int cur = n - 1; //当前节点，由于之前n++过了,所以减一
 //循环找到合适放tmp的位置，并不断向后移动元素给待放的tmp腾出位置
 while(cur > 0 && h[P(cur)] < tmp){  //当tmp比cur的父亲大的时候
  h[cur] = h[P(cur)];
  cur = P(cur);
 }
 //现在的cur位置就是合适放tmp的位置了
 h[cur] = tmp;
}

template
void Heap::heapify(int cur){
 T mmax = h[L(cur)] > h[R(cur)] ? h[L(cur)] : h[R(cur)];
 if(mmax < h[cur])
  return;
 //cout << "##########" << endl;
 //mmax.show();
 if(h[L(cur)] == mmax){
  h[0].swap(h[cur], h[L(cur)]);
  heapify(L(cur)); 
 }else{
  h[0].swap(h[cur], h[R(cur)]);
  heapify(R(cur));
 }
 //cout << "##########" << endl;

}

int main(){
 int num = 7;
 dNode d[M];
 for(int i = 0; i < num; i++){
  d[i] = dNode("Luo", rand() % 50, rand() % 11);
 }
 d[0].score = 10;
 d[1].score = 33;
 d[2].score = 22;
 d[3].score = 43;
 d[4].score = 7;
 d[5].score = 66;
 d[6].score = 1;

 Heap *h = new Heap(d, num);
 h->build();

 h->insert(d[1]);
 h->insert(d[3]);
 h->show();
 cout << "########### test top and pop ####" << endl;
 h->top().show();
 h->pop();
 h->top().show();

 return 0;
}