问题描述
如何打断像 [9,7,6,3,4,1,9]
-> [[9,3],[4,1],[7,[9]]
-> [25,5,10,9]
这样的向量?
其背后的逻辑是将向量分解为子向量,其中每个后续元素都小于前一个元素(0'z 被忽略),一个降序序列。当子向量形成时,每个子向量都被替换为其所有元素的总和。
[https://www.codewars.com/kata/5f8fb3c06c8f520032c1e091][1]
解决方法
迭代 nums
的元素以构建拆分。对于每个数字,将其与最后一个数字进行比较,以决定是创建子列表还是附加到现有的列表中:
let nums = vec![9,7,6,3,4,1,9];
let mut split: Vec<Vec<i32>> = vec![vec![]];
for num in nums.iter().filter(|n| **n != 0) {
let sublist = split.last_mut().unwrap();
match sublist.last_mut() {
Some(x) if num > x => {
split.push(vec![*num]);
}
_ => sublist.push(*num),}
}
let split = split; // make split immmutable
let summed: Vec<i32> = split.iter().map(|v| v.iter().sum()).collect();
可能可以使用 Iterator::partition_in_place
制定更优雅的解决方案,但遗憾的是,fn
目前还不稳定。