day_06.py

from collections import defaultdict, deque
from typing import TYPE_CHECKING

import aoc_helper
from aoc_helper import (
    Grid,
    PrioQueue,
    SparseGrid,
    decode_text,
    extract_ints,
    extract_iranges,
    extract_ranges,
    extract_uints,
    frange,
    irange,
    iter,
    list,
    map,
    range,
    search,
    tail_call,
)

raw = aoc_helper.fetch(6, 2023)


def parse_raw(raw: str):
    nums = extract_ints(raw)
    return nums.chunked(nums.len() // 2)


data = parse_raw(raw)


# providing this default is somewhat of a hack - there isn't any other way to
# force type inference to happen, AFAIK - but this won't work with standard
# collections (list, set, dict, tuple)
def part_one(data=data):
    total = 1
    for time, distance in zip(*data):
        wins = 0
        for hold in range(time):
            speed = hold * (time - hold)
            if speed > distance:
                wins += 1
        total *= wins
    return total


aoc_helper.lazy_test(day=6, year=2023, parse=parse_raw, solution=part_one)


# providing this default is somewhat of a hack - there isn't any other way to
# force type inference to happen, AFAIK - but this won't work with standard
# collections (list, set, dict, tuple)
def part_two(data=data):
    time = int("".join(map(str, data[0])))
    distance = int("".join(map(str, data[1])))
    wins = 0
    for hold in range(time):
        speed = hold * (time - hold)
        if speed > distance:
            wins += 1
    return wins


aoc_helper.lazy_test(day=6, year=2023, parse=parse_raw, solution=part_two)

aoc_helper.lazy_submit(day=6, year=2023, solution=part_one, data=data)
aoc_helper.lazy_submit(day=6, year=2023, solution=part_two, data=data)