From eac8ccf339716109de174f0287b84ae851fa4e4b Mon Sep 17 00:00:00 2001 From: Mylloon Date: Mon, 4 Dec 2023 16:15:17 +0100 Subject: [PATCH] q2 : easy --- 2023/day4/main.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/2023/day4/main.py b/2023/day4/main.py index fda7a3e..0b977cd 100644 --- a/2023/day4/main.py +++ b/2023/day4/main.py @@ -17,11 +17,21 @@ def part1(file: str): def part2(file: str): + array = [] with open(file) as f: for line in f: - pass + tmp = [] + for el in [i.split(" ") for i in line.split(":")[1].split("|")]: + tmp.append([int(it) for it in el if len(it) > 0]) + array.append(tmp) - return 0 + sets = [1 for _ in array] + for num_set, card in enumerate(array): + num_matches = len(set(card[0]) & set(card[1])) + for j in range(num_matches): + sets[num_set + 1 + j] += sets[num_set] + + return sum(sets) if __name__ == "__main__":