1. package com.atguigu.graph;
    3. import java.util.ArrayList;
    4. import java.util.Arrays;
    6. /**
    7.  * ClassName:  <br/>
    8.  * Description:  <br/>
    9.  * Date: 2021-03-04 10:36 <br/>
    10.  * @project data_algorithm
    11.  * @package com.atguigu.graph
    12.  */
    13. public class Graph {
    14.     private ArrayList<String> vertexList; // 存储定点集合
    15.     private int[][] edges; // 存储图对应的临界矩阵
    16.     private int numOfEdges; // 表示变的数目
    18.     public static void main(String[] args) {
    19.         // 测试一把图是否有创建ok
    20.         int n = 5;// 节点的个数
    21.         String vertexValue[] = {"A", "B", "C", "D", "E"};
    22.         // 创建图对象
    23.         Graph graph = new Graph(n);
    24.         // 循环添加顶点
    25.         for (String value : vertexValue) {
    26.             graph.insertVertex(value);
    27.         }
    28.         // 添加边
    29.         graph.insertEdge(0,1,1);// A-B
    30.         graph.insertEdge(0,2,1); // A-C
    31.         graph.insertEdge(1,2,1);  // B-C
    32.         graph.insertEdge(1,3,1);   // B-D
    33.         graph.insertEdge(1,4,1);    // B-E
    35.         graph.showGraph();
    36.     }
    37.     // 构造
    38.     public Graph(int n) {
    39.         // 初始化 矩阵和 vertexList
    40.         edges = new int[n][n];
    41.         vertexList = new ArrayList<String>();
    42.         numOfEdges = 0;
    43.     }
    45.     // 图中常用的方法
    46.     // 返回节点中的个数
    47.     public int getNumOfVertex() {
    48.         return vertexList.size();
    49.     }
    51.     // 显示图对应的矩阵
    52.     public void showGraph() {
    53.         for (int[] link : edges) {
    54.             System.out.println(Arrays.toString(link));
    55.         }
    56.     }
    57.     // 得到边的个数
    58.     public int getNumOfEdges() {
    59.         return numOfEdges;
    60.     }
    61.     // 返回节点i(下标)对应的数据 0->"A" 1->B 2->C
    62.     public String getValueByIndex(int i) {
    63.         return vertexList.get(i);
    64.     }
    66.     // 返回v1和v2的权值
    67.     public int getWeight(int v1, int v2) {
    68.         return edges[v1][v2];
    69.     }
    72.     /**
    73.      * 插入节点
    74.      * @param vertex
    75.      */
    76.     public void insertVertex(String vertex) {
    77.         vertexList.add(vertex);
    78.     }
    81.     /**
    82.      * 添加边
    83.      * @param v1    表示点的下标 即 是第几个顶点 "A"="B" "A"->0 "B" ->1
    84.      * @param v2    表示第二个顶点的下标
    85.      * @param weight 表示他们是用什么关联的 是0还是1
    86.      */
    87.     public void insertEdge(int v1, int v2, int weight) {
    88.         edges[v1][v2] = weight;
    89.         edges[v2][v1] = weight;
    90.     }
    93. }

    运行结果

    1. [0, 1, 1, 0, 0]
    2. [1, 0, 1, 1, 1]
    3. [1, 1, 0, 0, 0]
    4. [0, 1, 0, 0, 0]
    5. [0, 1, 0, 0, 0]
    7. Process finished with exit code 0