feat: add rust codes for space_complexity (#409)
parent
82bbdd444d
commit
2029d2b939
@ -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());
|
||||
}
|
Loading…
Reference in new issue