diff options
Diffstat (limited to '2021/12/puzzles.py')
| -rw-r--r-- | 2021/12/puzzles.py | 14 |
1 files changed, 7 insertions, 7 deletions
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__": |