package com.haiyang.datastructure.graph;
import java.util.ArrayList;
import java.util.Arrays;
public class GraphDemo {
public static void main(String[] args) {
int n = 5;
String[] vertexValues = new String[]{"A", "B", "C", "D", "E"};
Graph graph = new Graph(n);
for (String vertex : vertexValues) {
graph.insertVertex(vertex);
}
graph.insertEdge(0, 1, 1);
graph.insertEdge(0, 2, 1);
graph.insertEdge(1, 2, 1);
graph.insertEdge(1, 3, 1);
graph.insertEdge(1, 4, 1);
graph.showGraph();
}
}
class Graph {
private ArrayList<String> vertexList;
private int[][] edges;
private int numOfEdges;
public Graph(int n) {
edges = new int[n][n];
vertexList = new ArrayList<>(n);
numOfEdges = 0;
}
public int getNumOfVertex() {
return vertexList.size();
}
public void showGraph() {
for (int[] link : edges) {
System.out.println(Arrays.toString(link));
}
}
public void insertVertex(String vertex) {
vertexList.add(vertex);
}
public void insertEdge(int v1, int v2, int weight) {
edges[v1][v2] = weight;
edges[v2][v1] = weight;
numOfEdges++;
}
}