From 2918a33ca3255535a1fd268c37bec6119f590bf3 Mon Sep 17 00:00:00 2001 From: Tyrel Souza Date: Mon, 12 Dec 2022 12:09:28 -0500 Subject: [PATCH] print fun --- 2022/python/day12.py | 31 ++++++++++++++++++++++++------- 2022/python/matrix.py | 14 +++++++------- 2 files changed, 31 insertions(+), 14 deletions(-) diff --git a/2022/python/day12.py b/2022/python/day12.py index 2cbd557..f60eab0 100644 --- a/2022/python/day12.py +++ b/2022/python/day12.py @@ -7,6 +7,12 @@ criteria = lambda _cur, _neighbor: _neighbor - _cur <= 1 def build_graph(mx): + """ + This is the meat of the solution, + getting the valid neighbors - and then weighing those neighbors as pathable + - loop through each row/col and then find who that cell can navigate to, ULDR + - if it can navigate, add an edge + """ graph = Graph() for y, row in enumerate(mx): for x, _ in enumerate(row): @@ -19,39 +25,50 @@ def build_graph(mx): def part1(mx): start = matrix.find_in_matrix(mx, "S") end = matrix.find_in_matrix(mx, "E") + # Now that we got the start/end, fix those to 'a' and 'z' like the instructions said mx[start[0]][start[1]] = "a" mx[end[0]][end[1]] = "z" + # Turn the a-z into numbers 0-25 matrix.apply_to_all(mx, lambda x: ord(x) - ord("a")) graph = build_graph(mx) path = find_path(graph, start, end) - print(len(path.nodes)) + print(len(path.nodes) -1) def part2(mx): end = matrix.find_in_matrix(mx, "E") + # don't NEED this, because we go from all 'a', but need to replace it still s = matrix.find_in_matrix(mx, "S") + # Now that we got the start/end, fix those to 'a' and 'z' like the instructions said mx[s[0]][s[1]] = "a" mx[end[0]][end[1]] = "z" - starts = matrix.find_in_matrix(mx, "a", one=False) + + # Turn the a-z into numbers 0-25 matrix.apply_to_all(mx, lambda x: ord(x) - ord("a")) graph = build_graph(mx) - n_counts = [] + paths = [] + # Find the yx of all 'a' for start in starts: try: path = find_path(graph, start, end) - n_counts.append(len(path.nodes) - 1) + paths.append(path) except algorithm.NoPathError: + #print('x', end='') + # This 'a' is in a puddle of 'c's, so it's not pathable pass - print(n_counts) - print(min(n_counts)) + shortest = min(paths, key= lambda x: x.total_cost) + matrix.apply_to_all(mx, lambda x: chr(x+ord('a'))) + matrix.highlight(mx, red=shortest.nodes) + print(shortest.total_cost) def main(): mx = matrix.load_matrix_file(shared.get_fname(12), matrix.split_word_to_chr_list) - # part1(mx) + part1(mx) + mx = matrix.load_matrix_file(shared.get_fname(12), matrix.split_word_to_chr_list) part2(mx) diff --git a/2022/python/matrix.py b/2022/python/matrix.py index a67efa2..dc6c9ad 100644 --- a/2022/python/matrix.py +++ b/2022/python/matrix.py @@ -236,7 +236,7 @@ def pmx(*matrices, pad=True, space=True): print("".join([f(x) for x in c])) -def ppmx(*matrices, pad=True, space=True): +def ppmx(*matrices, pad=True, space=True, zero='.'): """ print a matrix of anything, Falsy values turns to `.` for clarity """ @@ -245,19 +245,19 @@ def ppmx(*matrices, pad=True, space=True): for row in matrices: r = [] for col in row: - r.append("".join([f"{x or '.'}".rjust(3) for x in col])) + r.append("".join([f"{x or zero}".rjust(3) for x in col])) print(" ".join(r)) else: for row in matrices: for c in row: if pad: - f = lambda x: f"{x or '.'}".rjust(2) + f = lambda x: f"{x or zero}".rjust(2) if space: - f = lambda x: f"{x or '.'}".rjust(3) + f = lambda x: f"{x or zero}".rjust(3) else: - f = lambda x: f"{x or '.'}" + f = lambda x: f"{x or zero}" if space: - f = lambda x: f"{x or '.'} " + f = lambda x: f"{x or zero} " print("".join([f(x) for x in c])) @@ -278,4 +278,4 @@ def highlight(matrix, red=[], green=[], blue=[]): for (y, x) in blue: new = f"{colors.BLUE}{mx[y][x]}{colors.ENDC}" mx[y][x] = new - ppmx(mx, pad=False, space=False) + ppmx(mx, pad=False, space=True, zero='0')