Files
advent-of-code/2020/day_6/custom_customs.py
2020-12-06 03:28:34 -03:00

34 lines
755 B
Python

with open('input') as f:
data = [line.strip() for line in f.read().split('\n\n')]
def solve_a(data):
count = 0
for group in data:
answers = set()
for c in group:
if c != '\n':
answers.add(c)
count += len(answers)
return count
def solve_b(data):
count = 0
for group in data:
count_map = {}
for c in group:
if c == '\n': continue
if c in count_map:
count_map[c] += 1
else:
count_map[c] = 1
for k in count_map:
if count_map[k] == len(group.split()): count += 1
print(count_map, count, len(group.split()))
return count
print(solve_a(data))
print(solve_b(data))