diff --git a/day-06/day-06.py b/day-06/day-06.py index dbf2ec5..427ef5b 100644 --- a/day-06/day-06.py +++ b/day-06/day-06.py @@ -6,8 +6,7 @@ def part_1(input): numbers = [int(n) for n in input[0].split(',')] count = [len([x for x in numbers if x == i]) for i in range(9)] for _ in range(80): - tmp = count[0] - count = [count[i+1] for i in range(8)] + tmp = count.pop(0) count[6] += tmp count.append(tmp) result = sum(count) @@ -18,8 +17,7 @@ def part_2(input): numbers = [int(n) for n in input[0].split(',')] count = [len([x for x in numbers if x == i]) for i in range(9)] for _ in range(256): - tmp = count[0] - count = [count[i+1] for i in range(8)] + tmp = count.pop(0) count[6] += tmp count.append(tmp) result = sum(count)