from dataclasses import dataclass

@dataclass
class Reindeer:
    name: str
    speed: int
    fly_time: int
    rest_time: int
    def distance_after(self, seconds):
        cycle_time = self.fly_time + self.rest_time
        cycles = seconds // cycle_time
        remaining_time = seconds % cycle_time
        if remaining_time > self.fly_time:
            remaining_time = self.fly_time
        return cycles * self.speed * self.fly_time + remaining_time * self.speed

def part1(reindeer: list[Reindeer]) -> int:
    return max(r.distance_after(2503) for r in reindeer)

def part2(reindeer: list[Reindeer]) -> int:
    points = [0] * len(reindeer)
    for s in range(1, 2504):
        distances = [r.distance_after(s) for r in reindeer]
        max_dist = max(distances)
        for i, d in enumerate(distances):
            if d == max_dist:
                points[i] += 1
    return max(points)

if __name__ == '__main__':
    reindeer = [
        Reindeer('Vixen', 8, 8, 53),
        Reindeer('Blitzen', 13, 4, 49),
        Reindeer('Rudolph', 20, 7, 132),
        Reindeer('Cupid', 12, 4, 43),
        Reindeer('Donner', 9, 5, 38),
        Reindeer('Dasher', 10, 4, 37),
        Reindeer('Comet', 3, 37, 76),
        Reindeer('Prancer', 9, 12, 97),
        Reindeer('Dancer', 37, 1, 36),
    ]
    print(part1(reindeer))
    print(part2(reindeer))