Re: Heap sorting in perl

[Abigail-II]

Yeah, whatever. Here's a modification of the program. Give the number of elements to extract as a command line argument, and the elements to extract from on STDIN. It assumes that there are enough elements on STDIN.

#!/usr/bin/perl

use strict;
use warnings;

my @heap;
my $M = @ARGV ? shift : 10;

sub heapify;
sub heapify {
    my $idx = shift;

    my $max = $idx;
    for my $try (2 * $idx + 1, 2 * $idx + 2) {
        $max = $try if $try < @heap && $heap [$try] > $heap [$max]
    }
    return if $max == $idx;

    @heap [$idx, $max] = @heap [$max, $idx];
    heapify $max;
}

sub extract_max {
    return unless @heap;
    my $min = $heap [0];
    my $tmp = pop @heap;
    if (@heap) {
        $heap [0] = $tmp;
        heapify 0;
    }
    return $min;
}

# First load the heap with initial data.
for (1 .. $M) {
    push @heap => scalar <>;
}
chomp @heap;

#
# Heapify it.
#
    
for (my $i = int (@heap / 2); $i --;) {
    heapify $i;
}
        
#
# Deal with the rest.
#
    
while (<>) {
    chomp;
    next if $_ >= $heap [0];  # It's too large.
    $heap [0] = $_;
    heapify 0;
}
    
my @result;
push @result => extract_max while @heap;
@result = reverse @result;
print "@result\n";
__END__

The running time of this is O (N log k), using O (k) memory.

Abigail

[Anonymous Monk]

Every so often I really wish that I had learned more about algorithms. This is one of those times.

Beautiful.