heapsort
を使用して実装するmin-heap
と、配列が最大から最小にソートされます。heapsort
これはusingの望ましい出力min-heap
ですか? heap
それ自体が最小から最大の構造を持っているため、ソートが完了した後に最小から最大に出力するために再度ソートするのは冗長に思えます。
コード:
#include <iostream>
#include <vector>
#include "random.h"
#include "print.h"
int parent(int i)
{
return (i - 1) / 2;
}
int left(int i)
{
if(i == 0)
return 1;
else
return 2*i;
}
int right(int i)
{ if(i == 0)
return 2;
else
return 2*i + 1;
}
void min_heapify(std::vector<int> &A, int i, int heapsize)
{
int smallest;
int l = left(i);
//std::cout << "left = " << l << std::endl;
int r = right(i);
//std::cout << "right = " << r << std::endl;
if(l <= heapsize && A[l] < A[i])
smallest = l;
else
smallest = i;
//std::cout << "smallest = " << smallest << std::endl;
if(r <= heapsize && A[r] < A[smallest])
smallest = r;
if(smallest != i) {
print(A);
exchange(A, i, smallest);
min_heapify(A, smallest, heapsize);
}
}
void build_min_heap(std::vector<int> &A)
{
int heapsize = A.size() - 1;
for(int i = (A.size() - 1) / 2; i >= 0; i--)
min_heapify(A, i, heapsize);
}
void heapsort(std::vector<int> &A)
{
int heapsize = A.size() - 1;
build_min_heap(A);
std::cout << "heapsort after buildmaxheap" << std::endl;
print(A);
for(int i = A.size() - 1; i > 0; i--) {
exchange(A, 0, i);
heapsize--;
std::cout << "heapsize = " << heapsize << std::endl;
min_heapify(A, 0, heapsize);
}
}
int main()
{
std::vector<int> B;
fill(B, 5);
print(B);
heapsort(B);
print(B);
return 0;
}
コードからの出力:
41 65 31 41 19
41 65 31 41 19
41 65 19 41 31
41 19 65 41 31
41 19 31 41 65
19 41 31 41 65
heapsort after buildmaxheap
19 31 41 41 65
heapsize = 3
65 31 41 41 19
31 65 41 41 19
heapsize = 2
heapsize = 1
65 41 41 31 19
heapsize = 0
65 41 41 31 19
20 要素の出力:
41 65 31 41 19 15 72 11 78 69 37 23 29 63 75 4 5 49 75 99
after buildmaxheap
4 5 15 11 19 23 29 41 31 69 37 41 72 63 75 65 78 49 75 99
after sort
99 78 75 75 72 69 65 63 49 41 41 37 31 29 23 19 15 11 5 4