advent-of-code/2021/d07/ex2/ex2.py

29 lines
592 B
Python
Raw Normal View History

2021-12-07 10:34:10 +01:00
#!/usr/bin/env python
import itertools
import sys
from math import ceil, floor
from typing import List
def solve(input: List[str]) -> int:
def fuel(dist: int) -> int:
return dist * (dist + 1) // 2
nums = list(int(n) for n in input[0].split(","))
mean = sum(nums) / len(nums)
upper = sum(fuel(abs(n - ceil(mean))) for n in nums)
lower = sum(fuel(abs(n - floor(mean))) for n in nums)
return min(upper, lower)
def main() -> None:
input = [line.strip() for line in sys.stdin.readlines()]
print(solve(input))
if __name__ == "__main__":
main()