Added (Weighted) Undirected graph

Python version 3

graphs/Directed (Weighted) Graph

@@ -96,12 +96,109 @@ class DirectedGraph:
 						visited.append(__[1])
 		return visited
 
-if __name__ == "__main__":
-	g = DirectedGraph()
-	# add 50 random nodes to the graph
-	g.fill_graph_randomly(50)
-	# you can add or remove any edge and vertex
-	g.add_pair(3, 5)
-	g.remove_pair(3,5)
-	g.dfs()
-	g.bgs()
+
+class Graph:
+	def __init__(self):
+		self.graph = {}
+
+	# adding vertices and edges
+	# adding the weight is optional
+	# handels repetition
+	def add_pair(self, u, v, w = 1):
+		# check if the u exists
+		if self.graph.get(u):
+			# if there already is a edge
+			if self.graph[u].count([w,v]) == 0:
+				self.graph[u].append([w, v])
+		else:
+			# if u does not exist
+			self.graph[u] = [[w, v]]
+		# add the other way
+		if self.graph.get(v):
+			# if there already is a edge
+			if self.graph[v].count([w,u]) == 0:
+				self.graph[v].append([w, u])
+		else:
+			# if u does not exist
+			self.graph[v] = [[w, u]]
+			
+	# handels if the input does not exist
+	def remove_pair(self, u, v):
+		if self.graph.get(u):
+			for _ in self.graph[u]:
+				if _[1] == v:
+					self.graph[u].remove(_)
+		# the other way round
+		if self.graph.get(v):
+			for _ in self.graph[v]:
+				if _[1] == u:
+					self.graph[v].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