package com.guigu.graph;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
/**
* @author: guorui fu
* @versiion: 1.0
* 图
*/
public class Graph {
private ArrayList<String> vertexList;//存储顶点集合
private int[][] edges;//存储节点对应的邻接矩阵
private int numOfEdges;//表示边的数目
//定义数组boolean[] 记录某个节点是否被访问过
private boolean[] isVisited ;
public static void main(String[] args) {
int n = 5;//节点的个数
String vertexValue[] = {"A","B","C","D","E"};
//创建图对象
Graph graph = new Graph(5);
//循环添加
for (String value : vertexValue){
graph.insertVertex(value);
}
//添加边 A-B A-C B-C B-D B-E
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();
//测试dfs遍历
// System.out.println("深度遍历");
// graph.dfs();
//测试广度优先
System.out.println("广度优先遍历");
graph.bfs();
}
//构造器
public Graph(int n){
//初始化矩阵和vertexList
edges = new int[n][n];
vertexList = new ArrayList<String>(n);
numOfEdges = 0;
isVisited = new boolean[5];
}
//得到第一个邻接节点的下标
/**
*
* @param index 遍历矩阵某index行
* @return 如果存在就返回对应的下标 否则返回-1
*/
public int getFirstNeighbor(int index){
for (int i = 0; i < vertexList.size(); i++) {
if (edges[index][i] > 0){
return i;
}
}
return -1;
}
//根据前一个邻接节点的下标获取下一个邻接节点下标
public int getNextNeighbor(int v1,int v2){
for (int i = v2 + 1; i < vertexList.size(); i++) {
if (edges[v1][i] > 0){
return i;
}
}
return -1;
}
//对一个节点广度优先遍历
private void bfs(boolean[] isVisited,int i){
int u;//表示队列头节点对应下标
int w;//邻接点w
//队列 记录节点访问的顺序
LinkedList queue = new LinkedList();
System.out.print(getValueByIndex(i) + "->");
isVisited[i] = true;
//节点加入列表
queue.addLast(i);
while (!queue.isEmpty()){
//取出头节点的下标
u = (Integer) queue.removeFirst();
//得到第一个邻接点的下标
w = getFirstNeighbor(u);
while (w != -1){//找到
if (!isVisited[w]){
System.out.print(getValueByIndex(w) + "->");
isVisited[w] = true;
//入队列
queue.addLast(w);
}
//以u去前节点,找w后面的下一个邻接点
w = getNextNeighbor(u,w);//体现出广度优先
}
}
}
//遍历所有的节点,都进行广度优先
public void bfs(){
for (int i = 0; i < getNumOfVertex(); i++) {
if (!isVisited[i]){
bfs(isVisited,i);
}
}
}
//深度优先算法 从i出发
private void dfs(boolean[] isVisited,int i){
//访问该节点 输出
System.out.print(getValueByIndex(i) + "->");
isVisited[i] = true;//该-节点设置成已访问
//查找节点i的第一个邻接点
int w = getFirstNeighbor(i);//查找与i邻接的
while (w != -1){
if (!isVisited[w]){
dfs(isVisited,w);
}
//如果w节点已经被访问过,去查找邻接节点的下一个邻接节点
w = getNextNeighbor(i,w);
}
}
//对dfs 进行重载,遍历所有的节点,进行dfs
public void dfs(){
//遍历所有的节点【回溯】
for (int i = 0; i < getNumOfVertex(); i++) {
if (!isVisited[i]){
dfs(isVisited,i);
}
}
}
//常用的方法
//返回节点的个数
public int getNumOfVertex(){
return vertexList.size();
}
//得到边的个数
public int getNumOfEdges(){
return numOfEdges;
}
//返回节点i对应的值 0->'A' 1->'B' 2->'C'
public String getValueByIndex(int i){
return vertexList.get(i);
}
//返回v1和v2的权值
public int getWeight(int v1, int v2){
return edges[v1][v2];
}
//显示图对应的矩阵
public void showGraph(){
for (int[] link:edges){
System.out.println(Arrays.toString(link));
}
}
//插入顶点
public void insertVertex(String vertex){
vertexList.add(vertex);
}
//添加边
/**
*无向图 所以edges[v1][v2]=edges[v2][v1]
* @param v1 表示点的下标 是第几个顶点
* @param v2 第二个顶点对应的下标
* @param weight 表示
*/
public void insertEdge(int v1,int v2,int weight){
edges[v1][v2] = weight;
edges[v2][v1] = weight;
numOfEdges++;
}
}
[数据结构] 图-多对多关系(深度优先遍历与广度优先遍历)
于 2023-03-27 00:13:14 首次发布