From e790654a6a8db28946ba16a3203feadb699e94ca Mon Sep 17 00:00:00 2001 From: Thomas Voss Date: Mon, 6 Dec 2021 09:38:35 +0100 Subject: Make code a little more efficient --- 2021/06/puzzles.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) (limited to '2021/06') diff --git a/2021/06/puzzles.py b/2021/06/puzzles.py index 93dcfea..9d3f2d4 100644 --- a/2021/06/puzzles.py +++ b/2021/06/puzzles.py @@ -14,15 +14,11 @@ def main() -> None: acc = collections.Counter(map(int, f.read().split(","))) for _ in range(DAYS): - counts: Counter[int] = collections.Counter() - for a, c in acc.items(): - if a: - counts[a - 1] += c - else: - counts[6] += c - counts[8] += c - - acc = counts + n = acc[0] + for i in range(8): + acc[i] = acc[i + 1] + acc[6] += n + acc[8] = n print(sum(acc.values())) -- cgit v1.2.3