Advertisement
Guest User

Untitled

a guest
Aug 5th, 2015
179
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.97 KB | None | 0 0
  1. #include <iostream>
  2.  
  3. const int d = 3;    /*Арность кучи*/
  4.  
  5. /*Реализация функций перестроения бинарной кучи*/
  6. template <class Item>
  7. void fixUp(Item a[], int node){
  8.     while(node > 1 && a[node/2] < a[node]){
  9.         std::swap(a[node], a[node/2]);
  10.         node = node / 2;
  11.     }
  12. }
  13.  
  14. template <class Item>
  15. void fixDown(Item a[], int node, int n){
  16.     while(2*node <= n){
  17.         int j = 2*node;
  18.  
  19.         if(j < n && a[j] < a[j + 1]) j++;
  20.         if(!(a[node] < a[j])) break;
  21.         std::swap(a[node], a[j]);
  22.         node = j;
  23.     }
  24. }
  25.  
  26. /*Реализация функций перестроения d-арной кучи*/
  27. /*
  28. first_child = node * d + 1 - (d - 1)
  29. last_child  = node * d + 1
  30. parent      = (node + d - 2) / d
  31. */
  32. template <class Item>
  33. void d_ary_fixUp(Item a[], int node){
  34.     while(node > 1 && a[(node + d - 2)/d] < a[node]){
  35.         std::swap(a[node], a[(node + d - 2)/d]);
  36.         node = (node + d - 2) / d;
  37.     }
  38. }
  39.  
  40. template <class Item>
  41. void d_ary_fixDown(Item a[], int node, int n){
  42.     int first_child = node*d + 1 - (d - 1);
  43.     int last_child = node*d + 1;
  44.     int max_child = first_child;    /*Начнем с самого первого*/
  45.  
  46.     while(d*node <= n){
  47.         /*Находим максимального потомка*/
  48.         for(int i = first_child; i <= last_child; i++)
  49.             if(a[i] > a[max_child])
  50.                 max_child = i;
  51.         /*Если он больше родителя, меняем их местами*/
  52.         if(a[max_child] > a[(max_child + d - 2) / d])
  53.             std::swap(a[max_child], a[(max_child + d - 2) / d]);
  54.         node = max_child;
  55.     }
  56. }
  57.  
  58. template <class Item>
  59. void d_ary_fixDown2(Item a[], int node, int n){
  60.     int first_child = node*d + 1 - (d - 1);
  61.     int last_child = node*d + 1;
  62.     int max_child = first_child;    /*Начнем с самого первого*/
  63.  
  64.     while(node*d + 1 - (d - 1) <= n){
  65.         int j = node*d + 1 - (d - 1);
  66.        
  67.         /*Ищем максимального потомка среди всех d потомков узла node*/
  68.         for(int i = first_child; i <= last_child; i++)
  69.             if(a[i] > a[max_child])
  70.                 max_child = i;
  71.         /*Если потомок max_child больше своего родителя, то меняем их местами*/
  72.         //if(a[max_child] > a[(max_child + d - 2) / d])
  73.         if(a[max_child] > a[node])
  74.             std::swap(a[max_child], a[(max_child + d - 2) / d]);
  75.  
  76.         node = j;
  77.         int first_child = node*d + 1 - (d - 1);
  78.         int last_child = node*d + 1;
  79.     }
  80. }
  81.  
  82. template <class Item>
  83. class PQ{
  84. private:
  85.     Item *pq;
  86.     int n;
  87. public:
  88.     PQ(int max){
  89.         pq = new Item[max + 1];
  90.         n = 0;
  91.     }
  92.  
  93.     ~PQ(){
  94.         delete[] pq;
  95.     }
  96.  
  97.     int empty() const{
  98.         return n == 0;
  99.     }
  100.  
  101.     void insert(Item item){
  102.         pq[++n] = item;
  103.         d_ary_fixUp(pq, n);
  104.     }
  105.  
  106.     Item getmax(){
  107.         if(!empty()){
  108.             std::swap(pq[1], pq[n]);
  109.             d_ary_fixDown2(pq, 1, n-1);
  110.             return pq[n--];
  111.         }else
  112.             return 0;
  113.     }
  114. };
  115.  
  116. int main(){
  117.     PQ<int> pq(50);
  118.     int p;
  119.  
  120.     for(int i = 0; i < 10; i++)
  121.         pq.insert(i);
  122.     for(int i = 0; i < 10; i++)
  123.         std::cout << pq.getmax() << " ";
  124.     std::cin >> p;
  125. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement