kamov/advent-of-code
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

[2021] solve day 5 python

Browse source
This commit is contained in:
Maciej Jur 2021-12-05 13:40:39 +01:00
parent fdb2825251
commit 51705d963f
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
2021/Python/day05.py
View file

@ -33,11 +33,11 @@ def find_points_in_line(x1, y1, x2, y2, skip_diagonal: bool = False) -> Iterable
return zip(range(x1, x2 + 1), range(y1, y2 + 1))
def solve1() -> int:
def solve1(skip_diagonal=True) -> int:
visited_pts = set()
crossed_pts = set()
for (start, end) in loader():
for x, y in find_points_in_line(*start, *end, skip_diagonal=True):
for x, y in find_points_in_line(*start, *end, skip_diagonal):
if (x, y) in visited_pts:
crossed_pts.add((x, y))
else:
@ -45,5 +45,10 @@ def solve1() -> int:
return len(crossed_pts)
def solve2() -> int:
return solve1(skip_diagonal=False)
if __name__ == '__main__':
print(solve1()) # 5690
print(solve2()) # 17741