Update grafos.py

Modificações para simplificação e melhora lógica na criação dos grafos.