vevurka

import unittest


class Graph(dict):
    def __init__(self, vertex_list, edge_list):
        for v in vertex_list:
            self[v] = set()
        for e in edge_list:
            self.add_edge(e)

vevurka

import unittest


class ListGraph(object):
    def __init__(self, number_of_vertices):
        self.number_of_vertices = number_of_vertices
        self.matrix = [[0] * number_of_vertices for _ in range(number_of_vertices)]

    def add_edge(self, v1, v2):
        self.matrix[v1][v2] = 1

vevurka

vevurka