#include <string>
#include "edge.h"
#ifndef GRAPH_H
#define GRAPH_H

class Graph
{
  public:
    //Graph();        // default constructor
    Graph(int num_nodes);

    ~Graph();       // destructor

    int getSize();       //get number of elements
    int getCapacity();   //get max number of nodes that can be fitted
    int add(Edge *edge);
    void remove(Vertex *row, Vertex *col);
    void displayMatrix();
    void displayChild(int stateNumber);
    void graph_auto();

    void printInfo();

  private:
    int size;
    int max_capacity;
    Edge **graph;

};
#endif GRAPH_H