From c93fc1b9a11c96a4554ec47029bf085597fdef60 Mon Sep 17 00:00:00 2001 From: Thomas Voss Date: Sun, 12 Dec 2021 06:46:33 +0100 Subject: Fix defaultdict typehint and use default func args --- 2021/12/puzzles.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to '2021/12/puzzles.py') diff --git a/2021/12/puzzles.py b/2021/12/puzzles.py index 8cfc506..2194c49 100644 --- a/2021/12/puzzles.py +++ b/2021/12/puzzles.py @@ -3,24 +3,24 @@ from collections import defaultdict -def solve(paths: defaultdict[list[str]], path: str) -> int: +def solve(paths: defaultdict[list[str]], path: str, flag: bool = False) -> int: acc = 0 - tokens = path[0].split(",") + tokens = path.split(",") for dest in paths[tokens[-1]]: if dest == "end": acc += 1 # START PART 1 elif not (dest.islower() and dest in tokens): - acc += solve(paths, (f"{path[0]},{dest}", False)) + acc += solve(paths, f"{path},{dest}") # END PART 1 START PART 2 elif dest != "start": if dest.islower() and dest in tokens: - if path[1]: + if flag: continue - acc += solve(paths, (f"{path[0]},{dest}", True)) + acc += solve(paths, f"{path},{dest}", True) else: - acc += solve(paths, (f"{path[0]},{dest}", path[1])) + acc += solve(paths, f"{path},{dest}", flag) # END PART 2 return acc @@ -34,7 +34,7 @@ def main() -> None: paths[x].append(y) paths[y].append(x) - print(solve(paths, ("start", False))) + print(solve(paths, "start", False)) if __name__ == "__main__": -- cgit v1.2.3