AdventOfCode2021/day13_task1.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
###############################################################################
# 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
    dots = fold(dots, data_set[i])

    return len(dots)


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