1  from lib import *
 2  
 3  input = read_input(2021, 6)
 4  
 5  
 6  nums = ints(input)
 7  nums = [*map(nums.count, range(9))]
 8  for _ in range(80):
 9      nums.append(x := nums.pop(0))
10      nums[6] += x
11  
12  print(sum(nums))
13  
14  
15  nums = ints(input)
16  nums = [*map(nums.count, range(9))]
17  for _ in range(256):
18      nums.append(x := nums.pop(0))
19      nums[6] += x
20  
21  print(sum(nums))