如何改进此功能:
use std::{env, process::exit};
fn get_grid() -> [[u8; 9]; 9] {
let mut grid: [[u8; 9]; 9] = Default::default();
let mut args: Vec<String> = env::args().collect();
if args.len() != 10 {
eprintln!("This program need 9 strings of 9 numbers between 0 and 9");
exit(1);
}
args.remove(0);
let _: Vec<_> = args
.iter()
.enumerate()
.map(|(i, arg)| {
let _line: Vec<_> = arg
.split(' ')
.enumerate()
.map(|(j, value)| match value.parse() {
Ok(x) => {
grid[i][j] = x;
x
}
Err(e) => {
eprintln!("Value {} is not a valid integer [{}]", value, e);
exit(1);
}
})
.collect();
})
.collect();
return grid;
}
据我了解,.map()
将在收集时构建一个新的可迭代对象(此处为Vec
)并返回它。我不需要具有此可迭代性,我只想修改一个外部数组,而无需通过此迭代构建任何东西。
在JavaScript中,有.map
,还有一个.forEach
,它在map
上进行迭代,但不返回任何内容。 Rust中有等效的东西吗?
我可能只使用for (index, value) in args.iter().enumerate()
,但我正在寻找一种避免显式循环(如果存在)的方法。
答案 0 :(得分:3)
要更改现有数据结构,使用显式循环是最惯用的方法:
for (i, arg) in args.iter().enumerate() {
for (j, value) in arg.split(' ').enumerate() {
match value.parse() {
Ok(x) => {
grid[i][j] = x;
}
Err(e) => {
eprintln!("Value {} is not a valid integer [{}]", value, e);
exit(1);
}
}
}
}
您可以使用Iterator::for_each
编写此代码,但是大多数Rust开发人员都不太可能认为它是“更好的”:
args.iter().enumerate().for_each(|(i, arg)| {
arg.split(' ')
.enumerate()
.for_each(|(j, value)| match value.parse() {
Ok(x) => {
grid[i][j] = x;
}
Err(e) => {
eprintln!("Value {} is not a valid integer [{}]", value, e);
exit(1);
}
})
});
无论使用哪种方式,都绝对都不应收集到随后丢弃的所有Vec
中。