AdventOfCode2021/day17_task2.py at master · HenrikSZ/AdventOfCode2021 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
###############################################################################
# Day x, Task y                                                               #
###############################################################################

import aoc_util


day = 17
data_str = """target area: x=20..30, y=-10..-5"""


def simulate(x_min, x_max, y_min, y_max, x_vel, y_vel):
    x = 0
    y = 0

    while 1:
        x += x_vel
        y += y_vel

        if (y_vel < 0 and y < y_min) or (x_vel > 0 and x > x_max) or (x_vel < 0 and x < x_min):
            return False

        if x >= x_min and x <= x_max and y <= y_max and y >= y_min:
            return True

        if x_vel > 0:
            x_vel -= 1
        if x_vel < 0:
            x_vel += 1

        y_vel -= 1


def task(data_set: list[str]) -> int:
    [_, coordinates] = data_set[0].split(": ")
    [x, y] = coordinates.split(", ")
    x = x[2:]
    [x1, x2] = x.split("..")
    y = y[2:]
    [y1, y2] = y.split("..")

    x1, x2, y1, y2 = int(x1), int(x2), int(y1), int(y2)

    x_min, x_max = min(x1, x2), max(x1, x2)
    y_min, y_max = min(y1, y2), max(y1, y2)

    count = 0
    for x_vel in range(-500, 500):
        for y_vel in range(-500, 500):
            next_y = simulate(x_min, x_max, y_min, y_max, x_vel, y_vel)
            if next_y:
                count += 1

    return count


aoc_util.run_with_data_str(task, data_str)
aoc_util.run_with_data_set(task, day)