blob: 327a78b1ae4b766eaa872534355bcab6e13696a5 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#!/usr/bin/env raku
sub MAIN (
Int $part where * == 1|2 = 1 #= part to run (1 or 2)
) {
my %bag;
my Int $count = 0;
my @valid_bags = "shiny gold",;
for "input".IO.lines -> $entry {
if $entry ~~ /^(.*) \s bags \s contain (.*)./ -> $match {
my $bag = $match[0];
for $match[1].split(", ") {
if $_ ~~ /(\d) \s (.*) \s bag/ -> $contain_bag {
push %bag{$bag}, {
count => $contain_bag[0].Int,
bag => $contain_bag[1].Str
};
}
}
}
}
if $part == 1 {
MAIN: while True {
my $previous_count = $count;
for keys %bag {
for @(%bag{$_}) -> $contain {
push @valid_bags, $_ if @valid_bags ∋ $contain<bag>;
}
}
$count = 0;
COUNT: for keys %bag {
for @(%bag{$_}) -> $contain {
if @valid_bags ∋ $contain<bag> {
$count++;
next COUNT;
}
}
}
last MAIN if $previous_count == $count;
}
} elsif $part == 2 {
$count = count-bags(%bag, 'shiny gold');
}
say "Part $part: ", $count;
}
# count-bags takes %bag & the bag for which we have to count the bags
# & returns the count. Count here means the number of bags that will
# be inside $bag.
sub count-bags (
%bag, Str $bag --> Int # count will be an integer.
) {
return 0 unless %bag{$bag};
my Int $count = 0;
for @(%bag{$bag}) {
my $sub_bags = count-bags(%bag, $_<bag>);
$count += $_<count>;
unless $sub_bags == 0 {
$count += $_<count> * $sub_bags;
}
}
return $count;
}
|