mirror of
https://github.com/krahets/hello-algo.git
synced 2026-04-28 04:20:44 +08:00
feat: add rust codes for graph chapter (#598)
* feat: add rust codes for graph chapter * update: remove Vertex and it's method to vertex.rs
This commit is contained in:
136
codes/rust/chapter_graph/graph_adjacency_matrix.rs
Normal file
136
codes/rust/chapter_graph/graph_adjacency_matrix.rs
Normal file
@@ -0,0 +1,136 @@
|
||||
/*
|
||||
* File: graph_adjacency_matrix.rs
|
||||
* Created Time: 2023-07-12
|
||||
* Author: night-cruise (2586447362@qq.com)
|
||||
*/
|
||||
|
||||
/* 基于邻接矩阵实现的无向图类型 */
|
||||
pub struct GraphAdjMat {
|
||||
// 顶点列表,元素代表“顶点值”,索引代表“顶点索引”
|
||||
pub vertices: Vec<i32>,
|
||||
// 邻接矩阵,行列索引对应“顶点索引”
|
||||
pub adj_mat: Vec<Vec<i32>>,
|
||||
}
|
||||
|
||||
impl GraphAdjMat {
|
||||
/* 构造方法 */
|
||||
pub fn new(vertices: Vec<i32>, edges: Vec<[usize; 2]>) -> Self {
|
||||
let mut graph = GraphAdjMat {
|
||||
vertices: vec![],
|
||||
adj_mat: vec![],
|
||||
};
|
||||
// 添加顶点
|
||||
for val in vertices {
|
||||
graph.add_vertex(val);
|
||||
}
|
||||
// 添加边
|
||||
// 请注意,edges 元素代表顶点索引,即对应 vertices 元素索引
|
||||
for edge in edges {
|
||||
graph.add_edge(edge[0], edge[1])
|
||||
}
|
||||
|
||||
graph
|
||||
}
|
||||
|
||||
/* 获取顶点数量 */
|
||||
pub fn size(&self) -> usize {
|
||||
self.vertices.len()
|
||||
}
|
||||
|
||||
/* 添加顶点 */
|
||||
pub fn add_vertex(&mut self, val: i32) {
|
||||
let n = self.size();
|
||||
// 向顶点列表中添加新顶点的值
|
||||
self.vertices.push(val);
|
||||
// 在邻接矩阵中添加一行
|
||||
self.adj_mat.push(vec![0; n]);
|
||||
// 在邻接矩阵中添加一列
|
||||
for row in &mut self.adj_mat {
|
||||
row.push(0);
|
||||
}
|
||||
}
|
||||
|
||||
/* 删除顶点 */
|
||||
pub fn remove_vertex(&mut self, index: usize) {
|
||||
if index >= self.size() {
|
||||
panic!("index error")
|
||||
}
|
||||
// 在顶点列表中移除索引 index 的顶点
|
||||
self.vertices.remove(index);
|
||||
// 在邻接矩阵中删除索引 index 的行
|
||||
self.adj_mat.remove(index);
|
||||
// 在邻接矩阵中删除索引 index 的列
|
||||
for row in &mut self.adj_mat {
|
||||
row.remove(index);
|
||||
}
|
||||
}
|
||||
|
||||
/* 添加边 */
|
||||
pub fn add_edge(&mut self, i: usize, j: usize) {
|
||||
// 参数 i, j 对应 vertices 元素索引
|
||||
// 索引越界与相等处理
|
||||
if i >= self.size() || j >= self.size() || i == j {
|
||||
panic!("index error")
|
||||
}
|
||||
// 在无向图中,邻接矩阵沿主对角线对称,即满足 (i, j) == (j, i)
|
||||
self.adj_mat[i][j] = 1;
|
||||
self.adj_mat[j][i] = 1;
|
||||
}
|
||||
|
||||
/* 删除边 */
|
||||
// 参数 i, j 对应 vertices 元素索引
|
||||
pub fn remove_edge(&mut self, i: usize, j: usize) {
|
||||
// 参数 i, j 对应 vertices 元素索引
|
||||
// 索引越界与相等处理
|
||||
if i >= self.size() || j >= self.size() || i == j {
|
||||
panic!("index error")
|
||||
}
|
||||
self.adj_mat[i][j] = 0;
|
||||
self.adj_mat[j][i] = 0;
|
||||
}
|
||||
|
||||
/* 打印邻接矩阵 */
|
||||
pub fn print(&self) {
|
||||
println!("顶点列表 = {:?}", self.vertices);
|
||||
println!("邻接矩阵 =");
|
||||
println!("[");
|
||||
for row in &self.adj_mat {
|
||||
println!(" {:?},", row);
|
||||
}
|
||||
println!("]")
|
||||
}
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fn main() {
|
||||
/* 初始化无向图 */
|
||||
// 请注意,edges 元素代表顶点索引,即对应 vertices 元素索引
|
||||
let vertices = vec![1, 3, 2, 5, 4];
|
||||
let edges = vec![[0, 1], [0, 3], [1, 2], [2, 3], [2, 4], [3, 4]];
|
||||
let mut graph = GraphAdjMat::new(vertices, edges);
|
||||
println!("\n初始化后,图为");
|
||||
graph.print();
|
||||
|
||||
/* 添加边 */
|
||||
// 顶点 1, 2 的索引分别为 0, 2
|
||||
graph.add_edge(0, 2);
|
||||
println!("\n添加边 1-2 后,图为");
|
||||
graph.print();
|
||||
|
||||
/* 删除边 */
|
||||
// 顶点 1, 3 的索引分别为 0, 1
|
||||
graph.remove_edge(0, 1);
|
||||
println!("\n删除边 1-3 后,图为");
|
||||
graph.print();
|
||||
|
||||
/* 添加顶点 */
|
||||
graph.add_vertex(6);
|
||||
println!("\n添加顶点 6 后,图为");
|
||||
graph.print();
|
||||
|
||||
/* 删除顶点 */
|
||||
// 顶点 3 的索引为 1
|
||||
graph.remove_vertex(1);
|
||||
println!("\n删除顶点 3 后,图为");
|
||||
graph.print();
|
||||
}
|
||||
Reference in New Issue
Block a user