diff --git a/src/dsa/various.py b/src/dsa/various.py index 4356039..030a5c3 100644 --- a/src/dsa/various.py +++ b/src/dsa/various.py @@ -48,17 +48,19 @@ def matrix_sum(matrix: list[list[int]]) -> list[int]: def graph_traversal(graph: dict[int, dict[int]], node: int) -> dict[int]: - visited = [] + visited = set() + result = [] def dfs(n: int) -> None: if n in visited: return - visited.append(n) - for neighbor in graph.get(n, []): + visited.add(n) + result.append(n) + for neighbor in graph.get(n, {}): dfs(neighbor) dfs(node) - return visited + return result def regex_match(strings: list[str], pattern: str) -> list[str]: