diff options
Diffstat (limited to '2020/aoc2020-d04.py')
| -rw-r--r-- | 2020/aoc2020-d04.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/2020/aoc2020-d04.py b/2020/aoc2020-d04.py new file mode 100644 index 0000000..22f044d --- /dev/null +++ b/2020/aoc2020-d04.py @@ -0,0 +1,42 @@ +#advent of code 2020 +#day 04 +import re + +PuzzleInput = open("04.in","r").read() +passports = PuzzleInput.split("\n\n"); +p1 = 0; +p2 = 0; +documents = dict(); +required = ["byr","iyr","eyr","hgt","hcl","ecl","pid"]; #puzzle description + +def part2(doc): + valid2 = True; + valid2 *= len(doc["byr"]) == 4 and int(doc["byr"]) >= 1920 and int(doc["byr"]) <= 2002; + valid2 *= len(doc["iyr"]) == 4 and int(doc["iyr"]) >= 2010 and int(doc["iyr"]) <= 2020; + valid2 *= len(doc["eyr"]) == 4 and int(doc["eyr"]) >= 2020 and int(doc["eyr"]) <= 2030; + valid2 *= doc["hgt"][-2:] == "cm" or doc["hgt"][-2:] == "in"; + if doc["hgt"][-2:] == "cm": + valid2 *= int(doc["hgt"][:-2]) >= 150 and int(doc["hgt"][:-2]) <= 193; + if doc["hgt"][-2:] == "in": + valid2 *= int(doc["hgt"][:-2]) >= 59 and int(doc["hgt"][:-2]) <= 76; + valid2 *= doc["hcl"][0] == "#"; + valid2 *= len(re.findall("[a-f]|[0-9]",doc["hcl"])) == 6; + valid2 *= doc["ecl"] in ["amb", "blu", "brn", "gry", "grn", "hzl", "oth"]; + valid2 *= len(doc["pid"]) == 9; + return valid2; + +for i,p in enumerate(passports): + p = p.replace("\n"," ") + documents[i] = dict(); + for entry in p.split(" "): + if len(entry) == 0: continue; + k,v = entry.split(":"); + documents[i][k] = v; + valid = True; + for req in required: + valid *= req in documents[i]; + p1 += 1*valid; + if valid: p2 += 1*part2(documents[i]); + +print("part 1 =",p1); +print("part 2 =",p2); |