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
|
from collections import Counter
from copy import copy
from dataclasses import dataclass
from typing import Iterator, List, Optional, Tuple, Callable
from aoc import BaseAssignment
Vector = Tuple[Tuple[int, int], Tuple[int, int]]
class Assignment(BaseAssignment):
def parse_item(self, item: str) -> Vector:
start, end = item.split('->')
return (
tuple(int(_) for _ in start.strip().split(',')),
tuple(int(_) for _ in end.strip().split(',')),
)
def read_input(self, example = False) -> List[Vector]:
return list(super().read_input(example))
def is_horizontal(v: Vector):
return v[0][0] == v[1][0]
def is_vertical(v: Vector):
return v[0][1] == v[1][1]
def points_in_vector(vector: Vector) -> List[Tuple[int, int]]:
first_x = min(vector[0][0], vector[1][0])
last_x = max(vector[0][0], vector[1][0])
first_y = min(vector[0][1], vector[1][1])
last_y = max(vector[0][1], vector[1][1])
return [
(vector[0][0], y)
for y
in range(first_y, last_y + 1)
] if is_horizontal(vector) else [
(x, vector[0][1])
for x
in range(first_x, last_x + 1)
] if is_vertical(vector) else []
class AssignmentOne(Assignment):
example_result = 5
def run(self, input: List[Vector]) -> int:
coordinates = []
for vector in input:
coordinates += points_in_vector(vector)
return len([c for c in Counter(coordinates).values() if c >= 2])
# class AssignmentTwo(Assignment):
# example_result = 1924
#
# def run(self, input: Tuple[List[int], List[BingoCard]]) -> int:
# nrs, cards = input
#
# in_game = copy(cards)
# for nr in nrs:
# for card in copy(in_game):
# card.mark(nr)
#
# if card.bingo:
# in_game.remove(card)
#
# if len(in_game) == 0:
# return nr * sum(card.unmarked)
|
