adventofcode2020/day6.py at main · pmadridb/adventofcode2020 · GitHub

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
with open("./input6.txt", encoding="utf-8") as file:
    answers_list = [l.rstrip("\n") for l in file]

def part1():
    unique_answers = 0
    answers = ''
    for line in answers_list:
        if not line:
            unique_answers+= len(list(dict.fromkeys(answers)))
            answers = ''
        else:
            answers = answers + line
    unique_answers+= len(list(dict.fromkeys(answers)))
    print(unique_answers)

def part2():
    unique_answers = 0
    answers = []
    for line in answers_list:
        if not line:
            unique_answers+= dupe_answers(answers)
            answers = []
        else:
            answers.append(line)
    unique_answers+= dupe_answers(answers)
    print(unique_answers)

def dupe_answers(answers):
    seen = {}
    for answer in answers:
        for char in answer:
            if char not in seen:
                seen[char] = 1
            else:
                seen[char] += 1
    return list(seen.values()).count(len(answers))

part1()
part2()