#! /usr/bin/env python3 from collections import Counter def part1(inp): groups = open(inp).read().split("\n\n") 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(inp): # number of questions for which everyone in a group answered 'yes' number_of_questions = 0 groups = open(inp).read().split("\n\n") 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__": part1("input.txt") part2("input.txt")