HashSet<Integer> getVertexGroup(int vertex) {
        for (HashSet<Integer> vertexGroup : vertexGroups) {
            if (vertexGroup.contains(vertex)) {
                return vertexGroup;
            }
        }
        return null;
    }