plais/advent-of-code-2021
1
0
Fork 0
Code Projects Releases Activity

Update day 6 solution
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pascal Lais
2021-12-06 07:58:40 +01:00
parent 445c24c043
commit c408a267a6
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
day-06/day-06.py
View File

@@ -6,8 +6,7 @@ def part_1(input):
numbers = [int(n) for n in input[0].split(',')] numbers = [int(n) for n in input[0].split(',')]
count = [len([x for x in numbers if x == i]) for i in range(9)] count = [len([x for x in numbers if x == i]) for i in range(9)]
for _ in range(80): for _ in range(80):
tmp = count[0] tmp = count.pop(0)
count = [count[i+1] for i in range(8)]
count[6] += tmp count[6] += tmp
count.append(tmp) count.append(tmp)
result = sum(count) result = sum(count)
@@ -18,8 +17,7 @@ def part_2(input):
numbers = [int(n) for n in input[0].split(',')] numbers = [int(n) for n in input[0].split(',')]
count = [len([x for x in numbers if x == i]) for i in range(9)] count = [len([x for x in numbers if x == i]) for i in range(9)]
for _ in range(256): for _ in range(256):
tmp = count[0] tmp = count.pop(0)
count = [count[i+1] for i in range(8)]
count[6] += tmp count[6] += tmp
count.append(tmp) count.append(tmp)
result = sum(count) result = sum(count)