AdventOfCode2021/day13_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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
###############################################################################
# Day x, Task y                                                               #
###############################################################################

import aoc_util


day = 13
data_str = """6,10
0,14
9,10
0,3
10,4
4,11
6,0
6,12
4,1
0,13
10,12
3,4
3,0
8,4
1,10
2,14
8,10
9,0

fold along y=7
fold along x=5"""


def fold(dots, instr):
    [axis, val] = instr.replace("fold along ", "").split("=")

    val = int(val)

    new_dots = set([])
    if axis == "x":
        for i in dots:
            if i[0] < val:
                new_dots.add(i)
            else:
                new_dots.add((val - (i[0] - val), i[1]))
    elif axis == "y":
        for i in dots:
            if i[1] < val:
                new_dots.add(i)
            else:
                new_dots.add((i[0], val - (i[1] - val)))

    return new_dots


def task(data_set: list[str]) -> int:
    dots = set([])
    i = 0
    while data_set[i] != "":
        [x, y] = data_set[i].split(",")
        dots.add((int(x), int(y)))
        i += 1

    i += 1

    while i < len(data_set):
        dots = fold(dots, data_set[i])
        i += 1

    dot = dots.pop()
    dots.add(dot)
    min_x = dot[0]
    max_x = dot[0]
    min_y = dot[1]
    max_y = dot[1]
    for i in dots:
        if i[0] < min_x:
            min_x = i[0]
        elif i[0] > max_x:
            max_x = i[0]

        if i[1] < min_y:
            min_y = i[1]
        elif i[1] > max_y:
            max_y = i[1]

    for y in range(min_y, max_y + 1):
        for x in range(min_x, max_x + 1):
            if (x, y) in dots:
                print("#", end="")
            else:
                print(" ", end="")

        print()
    return 0


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