-#!/usr/bin/perl -wn
+#!/usr/bin/perl -w
use strict;
+use POSIX;
our %want;
-BEGIN {
- my $want = shift @ARGV;
- my ($part,$knife) = $want =~ m/^([ha])(\d)$/ or die;
+our @part_order = qw(h b);
- $want{ $knife+1 } = $part =~ m/[h]/;
- $want{ $knife+4 } = $part =~ m/[ha]/;
+my $want = shift @ARGV;
+my ($wknife,$wparts) = $want =~ m/^(\d)([a-z]+)$/ or die;
+
+sub want ($) {
+ my ($colournum) = @_;
+ my $knife = ($colournum-1) % 3;
+ return 0 unless $knife == $wknife;
+ my $part = $part_order[ floor(($colournum-1) / 3) ];
+ die "huh colour $colournum?" unless defined $part;
+ return 0 unless $part =~ m/[$wparts]/o;
+ return 1;
}
our $drop;
-
-if (m/^\S/) {
- $drop =
- m/^2 / ||
- (m/^3 \d+ \d+ \d+ (\d+) / && !$want{$1});
+while (<>) {
+ if (m/^\S/) {
+ $drop =
+ m/^2 5 / ||
+ (m/^(?:3 1|2 3) \d+ \d+ (\d+) / && !want($1));
+ }
+ next if $drop;
+
+ print or die $!;
}
-
-next if $drop;
-
-print or die $!;