From 2029d2b9392554918eb1899759a0c00c7368515d Mon Sep 17 00:00:00 2001
From: sjinzh <99076655+sjinzh@users.noreply.github.com>
Date: Sun, 12 Mar 2023 15:16:35 +0800
Subject: [PATCH] feat: add rust codes for space_complexity (#409)

---
 codes/rust/Cargo.toml                         |   5 +
 .../space_complexity.rs                       | 109 ++++++++++++++++++
 codes/rust/include/include.rs                 |   5 +-
 3 files changed, 117 insertions(+), 2 deletions(-)
 create mode 100644 codes/rust/chapter_computational_complexity/space_complexity.rs

diff --git a/codes/rust/Cargo.toml b/codes/rust/Cargo.toml
index 396f68184..10209b5ba 100644
--- a/codes/rust/Cargo.toml
+++ b/codes/rust/Cargo.toml
@@ -14,6 +14,11 @@ path = "chapter_computational_complexity/time_complexity.rs"
 name = "worst_best_time_complexity"
 path = "chapter_computational_complexity/worst_best_time_complexity.rs"
 
+# Run Command: cargo run --bin space_complexity
+[[bin]]
+name = "space_complexity"
+path = "chapter_computational_complexity/space_complexity.rs"
+
 # Run Command: cargo run --bin leetcode_two_sum
 [[bin]]
 name = "leetcode_two_sum"
diff --git a/codes/rust/chapter_computational_complexity/space_complexity.rs b/codes/rust/chapter_computational_complexity/space_complexity.rs
new file mode 100644
index 000000000..3ab7bb23b
--- /dev/null
+++ b/codes/rust/chapter_computational_complexity/space_complexity.rs
@@ -0,0 +1,109 @@
+/*
+ * File: space_complexity.rs
+ * Created Time: 2023-03-11
+ * Author: sjinzh (sjinzh@gmail.com)
+ */
+
+include!("../include/include.rs");
+
+use std::collections::HashMap;
+use std::rc::Rc;
+use std::cell::RefCell;
+
+/* 函数 */
+fn function() ->i32 {
+    // do something
+    return 0;
+}
+
+/* 常数阶 */
+#[allow(unused)]
+fn constant(n: i32) {
+    // 常量、变量、对象占用 O(1) 空间
+    const A: i32 = 0;
+    let b = 0;
+    let nums = vec![0; 10000];
+    let node = ListNode::new(0);
+    // 循环中的变量占用 O(1) 空间
+    for i in 0..n {
+        let c = 0;
+    }
+    // 循环中的函数占用 O(1) 空间
+    for i in 0..n {
+        function();
+    }
+}
+
+/* 线性阶 */
+#[allow(unused)]
+fn linear(n: i32) {
+    // 长度为 n 的数组占用 O(n) 空间
+    let mut nums = vec![0; n as usize];
+    // 长度为 n 的列表占用 O(n) 空间
+    let mut nodes = Vec::new();
+    for i in 0..n {
+        nodes.push(ListNode::new(i))
+    }
+    // 长度为 n 的哈希表占用 O(n) 空间
+    let mut map = HashMap::new();
+    for i in 0..n {
+        map.insert(i, i.to_string());
+    }
+}
+
+/* 线性阶（递归实现） */
+fn linear_recur(n: i32) {
+    println!("递归 n = {}", n);
+    if n == 1 {return};
+    linear_recur(n - 1);
+}
+
+/* 平方阶 */
+#[allow(unused)]
+fn quadratic(n: i32) {
+    // 矩阵占用 O(n^2) 空间
+    let num_matrix = vec![vec![0; n as usize]; n as usize];
+    // 二维列表占用 O(n^2) 空间
+    let mut num_list = Vec::new();
+    for i in 0..n {
+        let mut tmp = Vec::new();
+        for j in 0..n {
+            tmp.push(0);
+        }
+        num_list.push(tmp);
+    }
+}
+
+/* 平方阶（递归实现） */
+fn quadratic_recur(n: i32) -> i32 {
+    if n <= 0 {return 0};
+    // 数组 nums 长度为 n, n-1, ..., 2, 1
+    let nums = vec![0; n as usize];
+    println!("递归 n = {} 中的 nums 长度 = {}", n, nums.len());
+    return quadratic_recur(n - 1);
+}
+
+/* 指数阶（建立满二叉树） */
+fn build_tree(n: i32) -> Option<Rc<RefCell<TreeNode>>> {
+    if n == 0 {return None};
+    let root = TreeNode::new(0);
+    root.borrow_mut().left = build_tree(n - 1);
+    root.borrow_mut().right = build_tree(n - 1);
+    return Some(root);
+}
+
+/* Driver Code */
+fn main() {
+    let n = 5;
+    // 常数阶
+    constant(n);
+    // 线性阶
+    linear(n);
+    linear_recur(n);
+    // 平方阶
+    quadratic(n);
+    quadratic_recur(n);
+    // 指数阶
+    let root = build_tree(n);
+    print_util::print_tree(&root.unwrap());   
+}
\ No newline at end of file
diff --git a/codes/rust/include/include.rs b/codes/rust/include/include.rs
index 8fadec40e..e1ffd8cb5 100644
--- a/codes/rust/include/include.rs
+++ b/codes/rust/include/include.rs
@@ -5,6 +5,7 @@
  */
 
 pub mod print_util;
-pub mod tree_node;
 pub mod list_node;
-pub use list_node::ListNode;
\ No newline at end of file
+pub use list_node::ListNode;
+pub mod tree_node;
+pub use tree_node::TreeNode;
\ No newline at end of file