Pagini recente » Cod sursa (job #3031848) | Cod sursa (job #27027) | Cod sursa (job #668324) | Cod sursa (job #829934) | Cod sursa (job #2797477)
from collections import defaultdict
class Graph:
def __init__(self, nodes):
self.numOfNodes = nodes
self.neighbours = defaultdict(list)
def addDirectedEdge(self, u, v):
self.neighbours[u].append(v)
def addUndirectedEdge(self, u, v):
self.neighbours[u].append(v)
self.neighbours[v].append(u)
def addTransposeEdge(self, u, v):
self.neighbours[v].append(u)
def deleteNodes(self, *nodes):
for node in nodes:
del self.neighbours[node]
self.numOfNodes -= 1
return self
def transpose(self):
if self.numOfNodes > 0:
transposed_graph = defaultdict(list)
for source, targets in self.neighbours.items():
for target in targets:
transposed_graph[target].append(source)
t = Graph(self.numOfNodes)
t.neighbours = transposed_graph
return t
else:
return None
def bfs(self, s):
visited = {i: False for i in range(1, self.numOfNodes + 1)}
queue = []
queue.append(s)
visited[s] = True
while queue:
s = queue.pop(0)
print(s)
for i in self.neighbours[s]:
if not visited[i]:
queue.append(i)
visited[i] = True
def topologicalSort(self, source, visited):
global result
visited[source] = True
result.append(source)
for i in self.neighbours[source]:
if not visited[i]:
self.topologicalSort(i, visited)
return result
file = 'sortaret.in'
with open(file, 'rt') as f:
content = f.readlines()
content = [line.strip().split() for line in content]
content = [[int(x) for x in line] for line in content]
N, M = content[0][0], content[0][1]
g = Graph(N)
for edges in content[1:]:
g.addDirectedEdge(edges[0], edges[1])
visited = {i: False for i in range(1, g.numOfNodes + 1)}
result = []
with open('sortaret.out', 'wt') as f:
for node in g.topologicalSort(1, visited):
f.write(str(node))
f.write(' ')