mirror of
https://github.com/thib8956/advent-of-code.git
synced 2025-08-23 16:01:59 +00:00
31 lines
838 B
Python
31 lines
838 B
Python
#! /usr/bin/env python3
|
|
from collections import Counter
|
|
|
|
|
|
def part1(groups):
|
|
number_of_questions = 0
|
|
for group in groups:
|
|
unique_questions = set(group.replace("\n", ""))
|
|
number_of_questions += len(unique_questions)
|
|
print(number_of_questions)
|
|
|
|
|
|
def part2(groups):
|
|
# number of questions for which everyone in a group answered 'yes'
|
|
number_of_questions = 0
|
|
for group in groups:
|
|
group_length = group.count("\n") + 1
|
|
group_counter = Counter(group.replace("\n", ""))
|
|
everyone_answered = [k for (k, v) in group_counter.items() if v == group_length]
|
|
number_of_questions += len(everyone_answered)
|
|
print(number_of_questions)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
inp = sys.argv[1]
|
|
groups = open(inp).read().split("\n\n")
|
|
part1(groups)
|
|
part2(groups)
|
|
|