如何修复Rust Trie实现中的“cannot borrow as mutable”错误？

核心错误修复 #

你的add_word方法存在两个关键问题：

1. 可变借用错误：通过&self.children[index]获取的是不可变引用，但调用add_word需要可变引用，触发编译错误。
2. 逻辑漏洞：子节点不存在时直接返回，而非创建新节点，导致无法添加新单词；此外，当s为空时设置is_end后未返回，后续代码会访问空字符串的字节索引，引发运行时panic。

修正后的add_word方法：

pub fn add_word(&mut self, s: &str) {
    if s.is_empty() {
        self.is_end = true;
        return; // 必须返回，避免后续越界访问
    }
    let index = (s.as_bytes()[0] - b'a') as usize;
    // 获取子节点的可变引用
    let child = &mut self.children[index];
    // 子节点不存在时创建新节点
    if child.is_none() {
        *child = Some(Trie::new());
    }
    // 可变借用子节点并递归调用
    child.as_mut().unwrap().add_word(s.get(1..).unwrap_or_default());
}

符合Rust惯用写法的优化建议 #

1. 用模式匹配替代unwrap，提升代码安全性 #

unwrap在生产环境易引发panic，推荐用if let或match安全处理Option：

优化后的starts_with：

pub fn starts_with(&self, s: &str) -> bool {
    if s.is_empty() {
        return true;
    }
    let index = (s.as_bytes()[0] - b'a') as usize;
    match &self.children[index] {
        Some(child) => child.starts_with(s.get(1..).unwrap_or_default()),
        None => false,
    }
}

优化后的add_word（用get_or_insert_with简化逻辑）：

pub fn add_word(&mut self, s: &str) {
    if s.is_empty() {
        self.is_end = true;
        return;
    }
    let index = (s.as_bytes()[0] - b'a') as usize;
    let child = self.children[index].get_or_insert_with(Trie::new);
    child.add_word(s.get(1..).unwrap_or_default());
}

2. 简化new方法 #

复用Default实现，让代码更简洁：

pub fn new() -> Box<Self> {
    Box::new(Self::default())
}

3. 用迭代替代递归，避免栈溢出风险 #

递归写法虽简洁，但单词过长时可能触发栈溢出，迭代写法更稳健且符合Rust风格：

pub fn add_word(&mut self, s: &str) {
    let mut current = self;
    for &byte in s.as_bytes() {
        let index = (byte - b'a') as usize;
        current = current.children[index].get_or_insert_with(Self::new).as_mut();
    }
    current.is_end = true;
}

pub fn starts_with(&self, s: &str) -> bool {
    let mut current = self;
    for &byte in s.as_bytes() {
        let index = (byte - b'a') as usize;
        match &current.children[index] {
            Some(child) => current = child.as_ref(),
            None => return false,
        }
    }
    true
}

4. 保持封装性，隐藏内部细节 #

移除children字段的pub修饰符，避免外部直接修改内部结构：

#[derive(Default)]
pub struct Trie {
    is_end: bool,
    children: [Option<Box<Trie>>; 26], // 去掉pub修饰符
}

完整优化后的代码 #

use std::array;

#[derive(Default)]
pub struct Trie {
    is_end: bool,
    children: [Option<Box<Trie>>; 26],
}

impl Trie {
    pub fn new() -> Box<Self> {
        Box::new(Self::default())
    }

    pub fn starts_with(&self, s: &str) -> bool {
        let mut current = self;
        for &byte in s.as_bytes() {
            let index = (byte - b'a') as usize;
            match &current.children[index] {
                Some(child) => current = child.as_ref(),
                None => return false,
            }
        }
        true
    }

    pub fn add_word(&mut self, s: &str) {
        let mut current = self;
        for &byte in s.as_bytes() {
            let index = (byte - b'a') as usize;
            current = current.children[index].get_or_insert_with(Self::new).as_mut();
        }
        current.is_end = true;
    }
}

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