comment on

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__
[download]

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

Abigail

In reply to Re: Heap sorting in perl by Abigail-II
in thread Heap sorting in perl by blakem

Are you posting in the right place? Check out Where do I post X? to know for sure.
Posts may use any of the Perl Monks Approved HTML tags. Currently these include the following:
<code> <a> <b> <big> <blockquote> <br /> <dd> <dl> <dt> <em> <font> <h1> <h2> <h3> <h4> <h5> <h6> <hr /> <i> <li> <nbsp> <ol> <p> <small> <strike> <strong> <sub> <sup> <table> <td> <th> <tr> <tt> <u> <ul>
Snippets of code should be wrapped in <code> tags not <pre> tags. In fact, <pre> tags should generally be avoided. If they must be used, extreme care should be taken to ensure that their contents do not have long lines (<70 chars), in order to prevent horizontal scrolling (and possible janitor intervention).
Want more info? How to link or or How to display code and escape characters are good places to start.


Think about Loose Coupling
	PerlMonks