Rust中如何以更地道方式优化嵌套while循环实现的twin_bound_zeros函数

你的代码功能正常，但手动管理索引、嵌套while循环加上标签的写法不够Rust化，可读性和简洁性都有提升空间。下面是用迭代器和模式匹配优化后的版本，逻辑更清晰，也更符合Rust的惯用写法：

fn twin_bound_zeros(fl: &[f64], start_idx: usize) -> Option<(usize, usize, f64)> {
    // 从指定起始索引开始迭代，同时获取索引和对应值
    let mut iter = fl.iter().enumerate().skip(start_idx);
    
    // 跳过开头连续的零值
    while let Some((i, &val)) = iter.next() {
        if val != 0.0 {
            // 把当前非零值的迭代位置放回去，后续要用到它作为左边界
            iter = fl.iter().enumerate().skip(i);
            break;
        }
    }

    // 用状态变量跟踪当前状态：左边界值、零区间起始索引
    let mut left_bound_val = None;
    let mut zero_interval_start = None;

    for (current_idx, &current_val) in iter {
        match (left_bound_val, zero_interval_start, current_val) {
            // 当前处于非零状态，第一次遇到零：记录零区间起始位置
            (Some(lv), None, 0.0) => {
                zero_interval_start = Some(current_idx);
            }
            // 当前处于零区间，遇到非零值：检查是否和左边界值相等
            (Some(lv), Some(zs), v) if v != 0.0 => {
                if lv == v {
                    return Some((zs, current_idx, v));
                } else {
                    // 不相等的话，更新左边界，重置零区间标记
                    left_bound_val = Some(v);
                    zero_interval_start = None;
                }
            }
            // 当前处于非零状态，遇到新的非零值：更新左边界
            (_, None, v) if v != 0.0 => {
                left_bound_val = Some(v);
            }
            // 其他情况（零区间内继续遇到零、开头没找到非零值等）：继续遍历
            _ => {}
        }
    }

    None
}

优化点说明 #

• 用enumerate()迭代器自动获取索引和值，避免手动增减索引，彻底杜绝越界风险
• 用match模式匹配替代嵌套while循环和标签，每个分支对应一种明确的状态转换，逻辑一目了然
• 用left_bound_val和zero_interval_start两个状态变量跟踪遍历状态，代码逻辑更直观
• 移除了原代码中l-1这类依赖前置条件的索引操作，通过状态跟踪确保左边界的有效性，代码更严谨

测试验证 #

保留你的测试用例，再补充几个场景验证正确性：

#[test]
fn works_for_nonzero_headed_arrays() {
    let float_list = [2.0, 0.0, 1.0, 0.0, 1.0];
    assert_eq!(twin_bound_zeros(&float_list, 0), Some((3, 4, 1.0)));
}

#[test]
fn handles_leading_zeros() {
    let float_list = [0.0, 0.0, 5.0, 0.0, 0.0, 5.0];
    assert_eq!(twin_bound_zeros(&float_list, 0), Some((3, 5, 5.0)));
}

#[test]
fn returns_none_when_no_match() {
    let float_list = [2.0, 0.0, 3.0, 0.0, 4.0];
    assert_eq!(twin_bound_zeros(&float_list, 0), None);
}

内容的提问来源于stack exchange，提问作者chart3r