mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-30 16:31:08 +00:00
Directed graph with optional weight assignment .
Containing graph auto-fill, dfs and bfs.
This commit is contained in:
parent
1c29a455b8
commit
fa2eecdc30
95
graphs/Directed (Weighted) Graph
Normal file
95
graphs/Directed (Weighted) Graph
Normal file
|
@ -0,0 +1,95 @@
|
||||||
|
from collections import deque
|
||||||
|
import random as rand
|
||||||
|
import math as math
|
||||||
|
|
||||||
|
# the dfault weight is 1 if not assigend but all the implementation is weighted
|
||||||
|
|
||||||
|
class DirectedGraph:
|
||||||
|
# enter True or False for this constructor
|
||||||
|
def __init__(self):
|
||||||
|
self.graph = {}
|
||||||
|
|
||||||
|
# adding vertices and edges
|
||||||
|
# note that self loops are not supported in undirected simpl graphs but it is in multigraphs
|
||||||
|
def add_pair(self, u, v, w = 1):
|
||||||
|
if self.graph.get(u):
|
||||||
|
if self.graph[u].count([w,v]) == 0:
|
||||||
|
self.graph[u].append([w, v])
|
||||||
|
else:
|
||||||
|
self.graph[u] = [[w, v]]
|
||||||
|
if not self.graph.get(v):
|
||||||
|
self.graph[v] = []
|
||||||
|
def remove_pair(self, u, v):
|
||||||
|
if self.graph.get(u):
|
||||||
|
for _ in self.graph[u]:
|
||||||
|
if _[1] == v:
|
||||||
|
self.graph[u].remove(_)
|
||||||
|
|
||||||
|
# if no destination is meant the defaut value is -1
|
||||||
|
def dfs(self, s = -2, d = -1):
|
||||||
|
if s == d:
|
||||||
|
return []
|
||||||
|
stack = []
|
||||||
|
visited = []
|
||||||
|
if s == -2:
|
||||||
|
s = list(self.graph.keys())[0]
|
||||||
|
stack.append(s)
|
||||||
|
visited.append(s)
|
||||||
|
ss = s
|
||||||
|
|
||||||
|
while True:
|
||||||
|
# check if there is any non isolated nodes
|
||||||
|
if len(self.graph[s]) != 0:
|
||||||
|
ss = s
|
||||||
|
for __ in self.graph[s]:
|
||||||
|
if visited.count(__[1]) < 1:
|
||||||
|
if __[1] == d:
|
||||||
|
visited.append(d)
|
||||||
|
return visited
|
||||||
|
else:
|
||||||
|
stack.append(__[1])
|
||||||
|
visited.append(__[1])
|
||||||
|
ss =__[1]
|
||||||
|
break
|
||||||
|
|
||||||
|
# check if all the children are visited
|
||||||
|
if s == ss :
|
||||||
|
stack.pop()
|
||||||
|
if len(stack) != 0:
|
||||||
|
s = stack[len(stack) - 1]
|
||||||
|
else:
|
||||||
|
s = ss
|
||||||
|
|
||||||
|
# check if se have reached the starting point
|
||||||
|
if len(stack) == 0:
|
||||||
|
return visited
|
||||||
|
|
||||||
|
# c is the count of nodes you want and if you leave it or pass -1 to the funtion the count
|
||||||
|
# will be random from 10 to 10000
|
||||||
|
def fill_graph_randomly(self, c = -1):
|
||||||
|
if c == -1:
|
||||||
|
c = (math.floor(rand.random() * 10000)) + 10
|
||||||
|
for _ in range(c):
|
||||||
|
# every vertex has max 100 edges
|
||||||
|
e = math.floor(rand.random() * 102) + 1
|
||||||
|
for __ in range(e):
|
||||||
|
n = math.floor(rand.random() * (c)) + 1
|
||||||
|
if n == _:
|
||||||
|
continue
|
||||||
|
self.add_pair(_, n, 1)
|
||||||
|
|
||||||
|
def bfs(self, s = -2):
|
||||||
|
d = deque()
|
||||||
|
visited = []
|
||||||
|
if s == -2:
|
||||||
|
s = list(self.graph.keys())[0]
|
||||||
|
d.append(s)
|
||||||
|
visited.append(s)
|
||||||
|
while d:
|
||||||
|
s = d.popleft()
|
||||||
|
if len(self.graph[s]) != 0:
|
||||||
|
for __ in self.graph[s]:
|
||||||
|
if visited.count(__[1]) < 1:
|
||||||
|
d.append(__[1])
|
||||||
|
visited.append(__[1])
|
||||||
|
return visited
|
Loading…
Reference in New Issue
Block a user