mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-10-06 13:49:30 +00:00
45 lines
1020 B
Python
45 lines
1020 B
Python
#!/usr/bin/python
|
||
|
||
# Author: OMKAR PATHAK
|
||
|
||
# We can use Python's dictionary for constructing the graph.
|
||
|
||
|
||
class AdjacencyList:
|
||
def __init__(self):
|
||
self.adj_list = {}
|
||
|
||
def add_edge(self, from_vertex: int, to_vertex: int) -> None:
|
||
# check if vertex is already present
|
||
if from_vertex in self.adj_list:
|
||
self.adj_list[from_vertex].append(to_vertex)
|
||
else:
|
||
self.adj_list[from_vertex] = [to_vertex]
|
||
|
||
def print_list(self) -> None:
|
||
for i in self.adj_list:
|
||
print((i, "->", " -> ".join([str(j) for j in self.adj_list[i]])))
|
||
|
||
|
||
if __name__ == "__main__":
|
||
al = AdjacencyList()
|
||
al.add_edge(0, 1)
|
||
al.add_edge(0, 4)
|
||
al.add_edge(4, 1)
|
||
al.add_edge(4, 3)
|
||
al.add_edge(1, 0)
|
||
al.add_edge(1, 4)
|
||
al.add_edge(1, 3)
|
||
al.add_edge(1, 2)
|
||
al.add_edge(2, 3)
|
||
al.add_edge(3, 4)
|
||
|
||
al.print_list()
|
||
|
||
# OUTPUT:
|
||
# 0 -> 1 -> 4
|
||
# 1 -> 0 -> 4 -> 3 -> 2
|
||
# 2 -> 3
|
||
# 3 -> 4
|
||
# 4 -> 1 -> 3
|