import networkx as nx
import matplotlib.pyplot as plt

G = nx.read_edgelist("lec01.data")

G.order()
G.size()
G.adj
G.nodes

nx.draw(G, with_labels=True)
plt.show()

counter = 0
S = {}
for v in G.nodes:
	S[v] = counter
	counter += 1

updates = 1
while updates > 0:
	updates = 0
	for v in G.nodes:
		for u in G.neighbors(v):
			if S[u] > S[v]:
				S[v] = S[u]
				updates += 1

colors = []
for v in G.nodes:
	if S[v] == 7:
		colors.append('green')
	elif S[v] == 5:
		colors.append('red')

nx.draw(G, with_labels=True, node_color=colors)
plt.show()