daviddoji/Advent_of_code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Solution to problem 5 part 2 in Python

Browse Source
This commit is contained in:
David Doblas Jiménez 2022-02-04 13:57:00 +01:00
parent b26e52c4ea
commit 4d5681a0be
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/Year_2021/P5.py
View File

@ -59,7 +59,7 @@
from collections import defaultdict from collections import defaultdict
from typing import DefaultDict, Tuple from typing import DefaultDict, Tuple
with open("files/test.txt") as f: with open("files/P5.txt") as f:
points = [ points = [
points points
for line in f.read().strip().split("\n") for line in f.read().strip().split("\n")
@ -139,6 +139,7 @@ def part_1() -> None:
def part_2() -> None: def part_2() -> None:
grid: DefaultDict[Tuple[int, int], int] = defaultdict(int) grid: DefaultDict[Tuple[int, int], int] = defaultdict(int)
for idx in range(len(points) // 2): for idx in range(len(points) // 2):
# print(idx)
if x_1[idx] == x_2[idx]: if x_1[idx] == x_2[idx]:
start, stop, step = get_range(y_1[idx], y_2[idx]) start, stop, step = get_range(y_1[idx], y_2[idx])
for pos in range(start, stop, step): for pos in range(start, stop, step):
@ -148,17 +149,16 @@ def part_2() -> None:
for pos in range(start, stop, step): for pos in range(start, stop, step):
grid[pos, y_1[idx]] += 1 grid[pos, y_1[idx]] += 1
elif abs(x_1[idx] - x_2[idx]) == abs(y_1[idx] - y_2[idx]): elif abs(x_1[idx] - x_2[idx]) == abs(y_1[idx] - y_2[idx]):
if x_2[idx] > x_1[idx]: start_x, stop_x, step_x = get_range(x_1[idx], x_2[idx])
start_x, stop_x, step_x = get_range(x_1[idx], x_2[idx]) start_y, stop_y, step_y = get_range(y_1[idx], y_2[idx])
start_y, stop_y, step_y = get_range(y_1[idx], y_2[idx]) pos_x = [pos_x for pos_x in range(start_x, stop_x, step_x)]
pos_x = [pos_x for pos_x in range(start_x, stop_x, step_x)] pos_y = [pos_y for pos_y in range(start_y, stop_y, step_y)]
pos_y = [pos_y for pos_y in range(start_y, stop_y, step_y)] for px, py in zip(pos_x, pos_y):
for px, py in zip(pos_x, pos_y): grid[px, py] += 1
grid[px, py] += 1
print(sum(1 for key in grid.values() if key > 1)) print(sum(1 for key in grid.values() if key > 1))
if __name__ == "__main__": if __name__ == "__main__":
# part_1() part_1()
part_2() part_2()