mirror of
https://github.com/krahets/hello-algo.git
synced 2024-12-26 11:16:28 +08:00
111 lines
3.2 KiB
Rust
111 lines
3.2 KiB
Rust
/*
|
||
* File: array.rs
|
||
* Created Time: 2023-01-15
|
||
* Author: xBLACICEx (xBLACKICEx@outlook.com), codingonion (coderonion@gmail.com)
|
||
*/
|
||
|
||
use hello_algo_rust::include::print_util;
|
||
use rand::Rng;
|
||
|
||
/* 随机访问元素 */
|
||
fn random_access(nums: &[i32]) -> i32 {
|
||
// 在区间 [0, nums.len()) 中随机抽取一个数字
|
||
let random_index = rand::thread_rng().gen_range(0..nums.len());
|
||
// 获取并返回随机元素
|
||
let random_num = nums[random_index];
|
||
random_num
|
||
}
|
||
|
||
/* 扩展数组长度 */
|
||
fn extend(nums: &[i32], enlarge: usize) -> Vec<i32> {
|
||
// 初始化一个扩展长度后的数组
|
||
let mut res: Vec<i32> = vec![0; nums.len() + enlarge];
|
||
// 将原数组中的所有元素复制到新
|
||
for i in 0..nums.len() {
|
||
res[i] = nums[i];
|
||
}
|
||
// 返回扩展后的新数组
|
||
res
|
||
}
|
||
|
||
/* 在数组的索引 index 处插入元素 num */
|
||
fn insert(nums: &mut [i32], num: i32, index: usize) {
|
||
// 把索引 index 以及之后的所有元素向后移动一位
|
||
for i in (index + 1..nums.len()).rev() {
|
||
nums[i] = nums[i - 1];
|
||
}
|
||
// 将 num 赋给 index 处的元素
|
||
nums[index] = num;
|
||
}
|
||
|
||
/* 删除索引 index 处的元素 */
|
||
fn remove(nums: &mut [i32], index: usize) {
|
||
// 把索引 index 之后的所有元素向前移动一位
|
||
for i in index..nums.len() - 1 {
|
||
nums[i] = nums[i + 1];
|
||
}
|
||
}
|
||
|
||
/* 遍历数组 */
|
||
fn traverse(nums: &[i32]) {
|
||
let mut _count = 0;
|
||
// 通过索引遍历数组
|
||
for i in 0..nums.len() {
|
||
_count += nums[i];
|
||
}
|
||
// 直接遍历数组元素
|
||
for num in nums {
|
||
_count += num;
|
||
}
|
||
}
|
||
|
||
/* 在数组中查找指定元素 */
|
||
fn find(nums: &[i32], target: i32) -> Option<usize> {
|
||
for i in 0..nums.len() {
|
||
if nums[i] == target {
|
||
return Some(i);
|
||
}
|
||
}
|
||
None
|
||
}
|
||
|
||
/* Driver Code */
|
||
fn main() {
|
||
/* 初始化数组 */
|
||
let arr: [i32; 5] = [0; 5];
|
||
print!("数组 arr = ");
|
||
print_util::print_array(&arr);
|
||
// 在 Rust 中，指定长度时（[i32; 5]）为数组，不指定长度时（&[i32]）为切片
|
||
// 由于 Rust 的数组被设计为在编译期确定长度，因此只能使用常量来指定长度
|
||
// Vector 是 Rust 一般情况下用作动态数组的类型
|
||
// 为了方便实现扩容 extend() 方法，以下将 vector 看作数组（array）
|
||
let nums: Vec<i32> = vec![1, 3, 2, 5, 4];
|
||
print!("\n数组 nums = ");
|
||
print_util::print_array(&nums);
|
||
|
||
// 随机访问
|
||
let random_num = random_access(&nums);
|
||
println!("\n在 nums 中获取随机元素 {}", random_num);
|
||
|
||
// 长度扩展
|
||
let mut nums: Vec<i32> = extend(&nums, 3);
|
||
print!("将数组长度扩展至 8 ，得到 nums = ");
|
||
print_util::print_array(&nums);
|
||
|
||
// 插入元素
|
||
insert(&mut nums, 6, 3);
|
||
print!("\n在索引 3 处插入数字 6 ，得到 nums = ");
|
||
print_util::print_array(&nums);
|
||
|
||
// 删除元素
|
||
remove(&mut nums, 2);
|
||
print!("\n删除索引 2 处的元素，得到 nums = ");
|
||
print_util::print_array(&nums);
|
||
|
||
// 遍历数组
|
||
traverse(&nums);
|
||
|
||
// 查找元素
|
||
let index = find(&nums, 3).unwrap();
|
||
println!("\n在 nums 中查找元素 3 ，得到索引 = {}", index);
|
||
}
|