import matrix
import shared
from dataclasses import dataclass
from functools import reduce # Valid in Python 2.6+, required in Python 3
import operator



@dataclass
class Game:
    duration: int
    highscore: int

# @shared.profile
def part1(rows):
    times = [int(r) for r in rows[0].split(":")[1].split(" ") if r]
    distances = [int(r) for r in rows[1].split(":")[1].split(" ") if r]
    ways = []
    for game in [Game(*g) for g in zip(times, distances)]:
        ways.append(calculate(game))
    print(reduce(operator.mul, ways, 1))


# @shared.profile
def part2(rows):
    # join ints
    times = [int("".join([r for r in rows[0].split(":")[1].split(" ") if r]))]
    distances = [int("".join([r for r in rows[1].split(":")[1].split(" ") if r]))]

    ways = []
    for game in [Game(*g) for g in zip(times, distances)]:
        ways.append(calculate(game))
    print(reduce(operator.mul, ways, 1))

def search(game, _range):
    for held in _range:
        remaining = game.duration - held
        score = held * remaining
        if score > game.highscore:
             return held   

def calculate(game):
    starting_win = search(game, range(game.duration+1))
    ending_win = search(game, reversed(range(game.duration+1)))
    print(game, starting_win, ending_win)
    return ending_win+1-starting_win

def main():
    rows = [row for row in shared.load_rows(6)]
    with shared.elapsed_timer() as elapsed:
        part1(rows)
        print("🕒", elapsed())

    rows = [row for row in shared.load_rows(6,True)]
    with shared.elapsed_timer() as elapsed:
        part2(rows)
        print("🕒", elapsed())


if __name__ == "__main__":
    main()