fork download
  1. #include <stdio.h>
  2. #include <stdlib.h>
  3.  
  4. #define height 4
  5. #define MAX (1<<height) //ビットシフト演算 2^height と同じ
  6.  
  7. int t[MAX+1]; //配列外アクセス防止のためのダミーで+1
  8. int sz = 0;
  9.  
  10. void swap(int *x, int *y){
  11. int tmp = *x;
  12. *x = *y;
  13. *y = tmp;
  14. }
  15.  
  16. void initTree(int n){
  17. int i;
  18. for(i=0;i<MAX;i++){
  19. t[i] = -1;
  20. }
  21. }
  22.  
  23. void printA(){
  24. int i;
  25. for(i=1;i<=sz;i++) printf("%d ",t[i]);
  26. printf("\n");
  27. }
  28.  
  29. void printT(int i){
  30. int x = i;
  31. while(x/2!=0){
  32. printf(" ");
  33. x/=2;
  34. }
  35. printf("%d\n",t[i]);
  36. }
  37.  
  38. int goP(int i){
  39. if(i/2 == 0) return 0;
  40. else return i/2;
  41. }
  42.  
  43. int goL(int i){
  44. if(2*i >= MAX) return 0;
  45. else return 2*i;
  46. }
  47.  
  48. int goR(int i){
  49. if(2*i+1 >= MAX) return 0;
  50. else return 2*i+1;
  51. }
  52.  
  53. void preOrder(int i){
  54. if(t[i] == -1) return;
  55. printT(i);
  56. preOrder(goL(i));
  57. preOrder(goR(i));
  58. }
  59.  
  60. void inOrder(int i){
  61. if(t[i] == -1) return;
  62. inOrder(goL(i));
  63. printT(i);
  64. inOrder(goR(i));
  65. }
  66.  
  67. void postOrder(int i){
  68. if(t[i] == -1) return;
  69. postOrder(goL(i));
  70. postOrder(goR(i));
  71. printT(i);
  72. }
  73.  
  74. void insBT(int x){
  75. int k,i = 1;
  76. for(k=0;k<height;k++){
  77. if(t[i]==-1){
  78. t[i] = x;
  79. sz++;
  80. return;
  81. }
  82. if(x < t[i]) i = goL(i);
  83. else i = goR(i);
  84. }
  85. printf("Error : too high -> %d\n",x);
  86. }
  87.  
  88.  
  89. int popHeap(){
  90. int i=1;
  91. int l,r,ma;
  92. int ret=t[i];
  93. t[i]=t[sz];
  94. t[sz--]=-1;
  95. while(i*2<=sz){
  96. l=goL(i);
  97. r=goR(i);
  98. if(t[l]<t[r]) ma=r;
  99. else ma=l;
  100. if(t[i]>t[ma]) break;
  101. swap(&t[i],&t[ma]);
  102. i=ma;
  103. }
  104. return ret;
  105. }
  106.  
  107. //末尾に要素を追加する
  108. //アップヒープ
  109. void pushHeap(int x){
  110. // 末尾に追加
  111. // 親と比較して、子の方が大きかったら交換
  112. // をHeapが完成するまで繰り返す
  113.  
  114.  
  115.  
  116.  
  117. }
  118.  
  119. int main(void){
  120. int i,x,n;
  121. scanf("%d",&n);
  122. // 木の初期化
  123. initTree(n);
  124. for(i=0;i<n;i++){
  125. scanf("%d",&t[i+1]);
  126. }
  127. sz = n;
  128. // 中間順で表示
  129. inOrder(1);
  130.  
  131. // pop
  132. int a=popHeap();
  133. printf("pop : %d\n",a);
  134. inOrder(1);
  135.  
  136. a = popHeap();
  137. printf("pop : %d\n",a);
  138.  
  139. inOrder(1);
  140.  
  141. printf("push : 100\n");
  142. pushHeap(100);
  143. inOrder(1);
  144.  
  145. printf("push : 30\n");
  146. pushHeap(30);
  147. inOrder(1);
  148. return 0;
  149. }
Success #stdin #stdout 0s 5284KB
stdin
12
89
34
55
13
16
21
32
2
3
4
5
8
stdout
      2
    13
      3
  34
      4
    16
      5
89
      8
    21
  55
    32
pop : 89
      2
    13
      3
  34
      4
    16
      5
55
    21
  32
    8
pop : 55
      2
    13
      3
  16
      4
    5
34
    21
  32
    8
push : 100
      2
    13
      3
  16
      4
    5
34
    21
  32
    8
push : 30
      2
    13
      3
  16
      4
    5
34
    21
  32
    8