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
Posts are HTML formatted. Put <p> </p> tags around your paragraphs. Put <code> </code> tags around your code and data!
Titles consisting of a single word are discouraged, and in most cases are disallowed outright.
Read Where should I post X? if you're not absolutely sure you're posting in the right place.
Please read these before you post! —
Posts may use any of the Perl Monks Approved HTML tags:
- a, abbr, b, big, blockquote, br, caption, center, col, colgroup, dd, del, div, dl, dt, em, font, h1, h2, h3, h4, h5, h6, hr, i, ins, li, ol, p, pre, readmore, small, span, spoiler, strike, strong, sub, sup, table, tbody, td, tfoot, th, thead, tr, tt, u, ul, wbr
You may need to use entities for some characters, as follows. (Exception: Within code tags, you can put the characters literally.)
|
For: |
|
Use: |
| & | | & |
| < | | < |
| > | | > |
| [ | | [ |
| ] | | ] |
Link using PerlMonks shortcuts! What shortcuts can I use for linking?
See Writeup Formatting Tips and other pages linked from there for more info.
|
|