-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDFS_BFS.txt
More file actions
44 lines (38 loc) · 1.26 KB
/
DFS_BFS.txt
File metadata and controls
44 lines (38 loc) · 1.26 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
from collections import deque
# Recursive DFS function
def dfs(node, visited, graph):
if node not in visited:
print(node, end=' ')
visited.add(node)
for neighbor in graph[node]:
dfs(neighbor, visited, graph)
# Recursive BFS function (using a helper function to simulate queue)
def bfs(queue, visited, graph):
if not queue:
return
node = queue.popleft()
if node not in visited:
print(node, end=' ')
visited.add(node)
for neighbor in graph[node]:
if neighbor not in visited:
queue.append(neighbor)
bfs(queue, visited, graph)
# -------- Input Graph --------
graph = {}
num_edges = int(input("Enter number of edges: "))
for _ in range(num_edges):
u, v = map(int, input("Enter edge (u v): ").split())
graph.setdefault(u, []).append(v)
graph.setdefault(v, []).append(u) # undirected
start = int(input("Enter starting node: "))
# -------- Traversals --------
# BFS Traversal (recursive)
print("BFS Traversal:", end=' ')
visited_bfs = set()
queue_bfs = deque([start])
bfs(queue_bfs, visited_bfs, graph)
# DFS Traversal (recursive)
print("\nDFS Traversal:", end=' ')
visited_dfs = set()
dfs(start, visited_dfs, graph)