Graph.java

package com.anuragkapur.ds.graph;

import java.util.ArrayList;
import java.util.List;

/**
 * @author: anuragkapur
 * @since: 09/05/2014
 */

public class Graph {

    List<Vertex> vertices;
    List<Edge> edges;

    public List<Vertex> getVertices() {
        return vertices;
    }

    public void setVertices(List<Vertex> vertices) {
        this.vertices = vertices;
    }

    public List<Edge> getEdges() {
        return edges;
    }

    public void setEdges(List<Edge> edges) {
        this.edges = edges;
    }

    public List<Edge> getEdgesFrom(Vertex v) {

        List<Edge> edgesFromVertex = new ArrayList<>();

        for (Edge edge : this.edges) {
            if (edge.getVertex1().equals(v)) {
                edgesFromVertex.add(edge);
            }
        }

        return edgesFromVertex;
    }
}