fishes = [list(map(int, open("input.txt").readline().split(","))).count(i) for i in range(9)]

for _ in range(256):  # For Part One: 80
    fishes = fishes[1:7] + [fishes[0] + fishes[7], fishes[8], fishes[0]]

print("Answer:", sum(fishes))

import aoc

import os
#import statistics
from typing import List, Callable
from functools import cache
import unittest

NUMBER = os.path.basename(__file__).removesuffix('.py')

#def task1_old(data: str) -> int:
#    data = [int(x) for x in data.split(',')]
#    median = int(statistics.median(data))
#    return sum(abs(position - median) for position in data)

def task(data: List[int], fuel: Callable[[int, int], int]) -> int:
    @cache
    def func(pos):
        return sum(fuel(crab, pos) for crab in data)
    pos = (data[-1]-data[0])//2
    step = (data[-1]-data[0])//4
    
    while True:
        dist = func(pos)
        dist_l = func(pos-1)
        dist_r = func(pos+1)
        if dist_l>=dist<=dist_r:
            return dist
        elif dist_r<dist:
            pos += step            
        else:
            pos -= step
        step = step//2 or 1        step = step//2 or 1

def task1(data: List[int]) -> int:
    return task(data, lambda crab, pos: abs(pos-crab))

def task2(data: List[int]) -> int:
    return task(data, lambda crab, pos: abs(pos-crab)*(abs(pos-crab)+1)//2)

class Test(unittest.TestCase):
    def test_task1(self):
        data = sorted(int(x) for x in [16,1,2,0,4,2,7,1,2,14])
        self.assertEqual(task1(data),37)
    def test_task2(self):
        data = sorted(int(x) for x in [16,1,2,0,4,2,7,1,2,14])
        self.assertEqual(task2(data),168)

if __name__ == '__main__':
    #unittest.main()
    data = aoc.load_data(NUMBER, verbose=True)
    data = sorted(int(x) for x in data.split(','))
    print('Task1:', task1(data))
    print('Task2:', task2(data))

import re

data = list()

with open("input.txt") as file:
    for line in file:
        entry = ["".join(sorted(i)) for i in re.findall(r"\w+", line)]
        data.append((entry[:10], entry[10:]))

#  tttt
# l    r
# t    t
#  mmmm
# l    r
# b    b
#  bbbb

count = 0
sum = 0

for patterns, outputs in data:
    digits = dict()
    fives = list()
    sixs = list()

    # get basic numbers as 1, 4, 7, 8
    # and collect digits by segments
    for pattern in patterns:
        if len(pattern) == 2:
            digits["1"] = list(pattern)
        elif len(pattern) == 4:
            digits["4"] = list(pattern)
        elif len(pattern) == 3:
            digits["7"] = list(pattern)
        elif len(pattern) == 7:
            digits["8"] = list(pattern)
        elif len(pattern) == 5:
            fives.append(list(pattern))
        else:
            sixs.append(list(pattern))

    for one in digits["1"]:
        # get 6, rt
        for six in sixs[:]:
            if one not in six:
                digits["6"] = six
                rt = one
                sixs.remove(six)
                break
        if len(six) == 2:
            break

    # get rb
    for one in digits["1"]:
        if one is not rt:
            rb = one
            break

    # get 2, 3, 5
    for one in digits["1"]:
        for five in fives:
            if rt not in five:
                digits["5"] = five
            elif rb not in five:
                digits["2"] = five
            else:
                digits["3"] = five

    lb = (set(digits["6"]) - set(digits["5"])).pop()

    # get 0, 9
    for six in sixs:
        if lb in six:
            digits["0"] = six
        else:
            digits["9"] = six

    digits = dict(zip(["".join(i) for i in digits.values()], digits.keys()))
    numbers = str()

    for output in outputs:
        if digits[output] in ["1", "4", "7", "8"]:
            count += 1
        numbers += digits[output]
    sum += int(numbers)

print("Part One:", count)
print("Part Two:", sum)

import aoc

import os

NUMBER = os.path.basename(__file__).removesuffix('.py')

def task1(data: str) -> int:
    data = [[int(height) for height in line] for line in data.splitlines()]
    result = 0
    for i, line in enumerate(data):
        for j, height in enumerate(line):
            if (    (i==0              or height<data[i-1][j])
                and (j==0              or height<data[i][j-1])
                and (i==len(data)-1    or height<data[i+1][j])
                and (j==len(data[0])-1 or height<data[i][j+1])
                ):
                result += height+1
    return result

def floodfill(cave, y, x):
    if y<0 or x<0 or y==len(cave) or x==len(cave[0]) or cave[y][x]==9:
        return 0
    cave[y][x] = 9
    return (  floodfill(cave, y+1, x) + floodfill(cave, y-1, x)
            + floodfill(cave, y, x+1) + floodfill(cave, y, x-1)) + 1

def task2(data: str) -> int:
    data = [[int(height) for height in line] for line in data.splitlines()]
    basins = []
    for i in range(len(data)):
        for j in range(len(data[0])):
            basins.append(floodfill(data, i, j))
    basins.sort()
    return basins[-1]*basins[-2]*basins[-3]
    

if __name__ == '__main__':
    data = aoc.load_data(NUMBER, verbose=True)
    print('Task1:', task1(data))
    print('Task2:', task2(data))