33 lines
886 B
Python
33 lines
886 B
Python
|
from collections import Counter
|
||
|
|
|
||
|
|
from solutions import BaseSolution
|
||
|
|
|
||
|
|
|
||
|
|
class Solution(BaseSolution):
|
||
|
|
input_file = "06.txt"
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return "Day 6: Custom Customs"
|
||
|
|
|
||
|
|
def parse_input(self, data):
|
||
|
|
return [g.split() for g in data.split("\n\n")]
|
||
|
|
|
||
|
|
def solve(self, puzzle_input):
|
||
|
|
return sum(self.count_any_yes_answers(g) for g in puzzle_input)
|
||
|
|
|
||
|
|
def solve_again(self, puzzle_input):
|
||
|
|
return sum(self.count_every_yes_answers(g) for g in puzzle_input)
|
||
|
|
|
||
|
|
def count_any_yes_answers(self, groupdata):
|
||
|
|
return len(set("".join(groupdata)))
|
||
|
|
|
||
|
|
def count_every_yes_answers(self, groupdata):
|
||
|
|
groupsize = len(groupdata)
|
||
|
|
answer_counts = Counter("".join(groupdata))
|
||
|
|
return sum(v == groupsize for _k, v in answer_counts.items())
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
solution = Solution()
|
||
|
|
solution.show_results()
|