2023/day06.py

import fileinput
from math import ceil, floor, sqrt

def race(time, dist):
    discr = sqrt(time * time - 4 * dist)
    x1 = (time - discr) / 2
    x2 = (time + discr) / 2
    x1, x2 = floor(x1) + 1, ceil(x2) - 1
    return x2 - x1 + 1

def main():
    input = list(fileinput.input())
    times = [int(x) for x in input[0][11:].split()]
    distances = [int(x) for x in input[1][11:].split()]

    part1 = 1
    for time, dist in zip(times, distances):
        part1 *= race(time, dist)

    time = int(input[0][11:].replace(" ", ""))
    dist = int(input[1][11:].replace(" ", ""))
    part2 = race(time, dist)

    print(part1, part2)

main()