-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path04.koto
140 lines (118 loc) · 4.03 KB
/
04.koto
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
# https://adventofcode.com/2020/day/4
expected_fields = "byr", "iyr", "eyr", "hgt", "hcl", "ecl", "pid"
eye_colours = "amb", "blu", "brn", "gry", "grn", "hzl", "oth"
parse_passports = |passports|
result = []
passport = {}
for line in passports.lines()
if line.is_empty()
result.push passport
passport = {}
for field in line.split(" ")
if field.is_empty()
continue
key, value = field.split(":").to_tuple()
if key == "cid" or expected_fields.contains key
passport.insert key, value
else
assert false
result.push passport
digits = "0", "1", "2", "3", "4", "5", "6", "7", "8", "9"
is_digit = |c| digits.contains c
passport_is_valid_part_one = |passport|
expected_fields.all |field|
passport.contains_key field
passport_is_valid_part_two = |passport|
for field in expected_fields
value = passport.get field
if not value then return false
match field
"byr" then if not 1920 <= value.to_number() <= 2002 then return false
"iyr" then if not 2010 <= value.to_number() <= 2020 then return false
"eyr" then if not 2020 <= value.to_number() <= 2030 then return false
"hgt" then
end_of_number = value.position |c| not is_digit c
if not end_of_number then return false
n = value[..end_of_number].to_number()
unit = value[end_of_number..]
match unit
"cm" then if not 150 <= n <= 193 then return false
"in" then if not 59 <= n <= 76 then return false
_ then return false
"hcl" then
if value[0] != "#" then return false
color = value[1..]
if color.size() != 6 then return false
if not color.all(|c| "0123456789abcdef".contains c) then return false
"ecl" then if not (eye_colours.contains value) then return false
"pid" then
if value.size() != 9 then return false
if not value.all(|digit| is_digit digit) then return false
true
@main = ||
passports = io.extend_path koto.script_dir, "input", "04"
>> io.read_to_string
>> parse_passports
part_one = passports
.keep |passport| passport_is_valid_part_one passport
.count()
print "Part one: $part_one" # 233
part_two = passports
.keep |passport| passport_is_valid_part_two passport
.count()
print "Part two: $part_two" # 111
@tests =
@pre_test: ||
self.passports = parse_passports "\
ecl:gry pid:860033327 eyr:2020 hcl:#fffffd
byr:1937 iyr:2017 cid:147 hgt:183cm
iyr:2013 ecl:amb cid:350 eyr:2023 pid:028048884
hcl:#cfa07d byr:1929
hcl:#ae17e1 iyr:2013
eyr:2024
ecl:brn pid:760753108 byr:1931
hgt:179cm
hcl:#cfa07d eyr:2025 pid:166559648
iyr:2011 ecl:brn hgt:59in
"
@test parse_passports: ||
assert_eq self.passports.size(), 4
assert_eq self.passports[0].ecl, "gry"
assert_eq self.passports[1].byr, "1929"
assert_eq self.passports[2].hgt, "179cm"
assert_eq self.passports[3].eyr, "2025"
@test part_one: ||
assert passport_is_valid_part_one self.passports[0]
assert not passport_is_valid_part_one self.passports[1]
assert passport_is_valid_part_one self.passports[2]
assert not passport_is_valid_part_one self.passports[3]
@test part_two_valid: ||
passports = parse_passports "\
pid:087499704 hgt:74in ecl:grn iyr:2012 eyr:2030 byr:1980
hcl:#623a2f
eyr:2029 ecl:blu cid:129 byr:1989
iyr:2014 pid:896056539 hcl:#a97842 hgt:165cm
hcl:#888785
hgt:164cm byr:2001 iyr:2015 cid:88
pid:545766238 ecl:hzl
eyr:2022
iyr:2010 hgt:158cm hcl:#b6652a ecl:blu byr:1944 eyr:2021 pid:093154719
"
for passport in passports
if not passport_is_valid_part_two passport
debug passport
assert false
@test part_two_invalid: ||
passports = parse_passports "\
eyr:1972 cid:100
hcl:#18171d ecl:amb hgt:170 pid:186cm iyr:2018 byr:1926
iyr:2019
hcl:#602927 eyr:1967 hgt:170cm
ecl:grn pid:012533040 byr:1946
hcl:dab227 iyr:2012
ecl:brn hgt:182cm pid:021572410 eyr:2020 byr:1992 cid:277
hgt:59cm ecl:zzz
eyr:2038 hcl:74454a iyr:2023
pid:3556412378 byr:2007
"
assert passports.all |passport| not passport_is_valid_part_two passport