Question

我想重构此Rust代码以计算最大系列产品，并使其尽可能高效和美观。我觉得

lsp(string_digits: &str, span: usize) -> Result<u64, Error>

可以通过某种方式使其变得比现在优雅得多。只能用一系列的链式迭代器方法来实现lsp吗？

#[derive(Debug, PartialEq)]
pub enum Error {
    SpanTooLong,
    InvalidDigit(char),
}

fn sp(w: &[u8]) -> u64 {
    w.iter().fold(1u64, |acc, &d| acc * u64::from(d))
}

pub fn lsp(string_digits: &str, span: usize) -> Result<u64, Error> {
    let invalid_chars = string_digits
        .chars()
        .filter(|ch| !ch.is_numeric())
        .collect::<Vec<_>>();
    if span > string_digits.len() {
        return Err(Error::SpanTooLong);
    } else if !invalid_chars.is_empty() {
        return Err(Error::InvalidDigit(invalid_chars[0]));
    } else if span == 0 || string_digits.is_empty() {
        return Ok(1);
    }

    let vec_of_u8_digits = string_digits
        .chars()
        .map(|ch| ch.to_digit(10).unwrap() as u8)
        .collect::<Vec<_>>();
    let lsp = vec_of_u8_digits
        .windows(span)
        .max_by(|&w1, &w2| sp(w1).cmp(&sp(w2)))
        .unwrap();
    Ok(sp(lsp))
}

Answer 1

不确定这是否是最优雅的方法，但是我已经尝试过，希望新版本等同于给定的程序。

在这种情况下，需要做两件事：首先，我们需要一个数据结构，该数据结构提供“即时”滑动窗口，其次，如果转换产生错误，则该函数需要一个提前结束迭代的函数。

对于前者，我选择了VecDeque，因为span是动态的。对于后者，在itertools板条箱中有一个名为process_results的函数。它将结果上的迭代器转换为未包装类型上的迭代器，并在遇到错误时停止迭代。

我还略微更改了sp的签名，以接受u8上的任何迭代器。

这是代码：

use std::collections::VecDeque;
use itertools::process_results;

#[derive(Debug, PartialEq)]
pub enum Error {
    SpanTooLong,
    InvalidDigit(char),
}

fn sp(w: impl Iterator<Item=u8>) -> u64 {
    w.fold(1u64, |acc, d| acc * u64::from(d))
}

pub fn lsp(string_digits: &str, span: usize) -> Result<u64, Error> {
    if span > string_digits.len() {
        return Err(Error::SpanTooLong);
    } else if span == 0 || string_digits.is_empty() {
        return Ok(1);
    }

    let mut init_state = VecDeque::new();
    init_state.resize(span, 0);

    process_results(string_digits.chars()
        .map(|ch| ch.to_digit(10)
            .map(|d| d as u8)
            .ok_or(Error::InvalidDigit(ch))),
        |digits|
            digits.scan(init_state, |state, digit| {
                state.pop_back();
                state.push_front(digit);
                Some(sp(state.iter().cloned()))
            })
            .max()
            .unwrap()
    )
}

我需要重构帮助以解决Rust中的错误

1 个答案: