优化Rust字符串处理函数的堆内存分配问题求助

原代码的内存分配问题 #

你的代码堆分配次数过多，核心原因有这几点：

• 每个单词处理时都会创建first_and_dot临时字符串，再通过+=追加到结果中，每次追加都可能触发String的扩容（扩容会重新分配堆内存并复制已有内容）
• 最后调用trim_end().to_string()会额外生成一个新String，相当于把已经处理好的内容再完整复制一遍
• 初始的one_text_initials是空字符串，后续追加过程中会多次触发扩容分配

优化方向 #

1. 预计算容量，避免频繁扩容：先算出每个结果字符串的精确长度，用String::with_capacity一次性分配足够内存，从根源减少扩容次数
2. 砍掉临时字符串：直接把首字母和". "写入目标String，别先创建临时字符串再追加
3. 去掉冗余复制：控制空格的添加时机，避免末尾出现空格，这样就不用调用trim_end().to_string()做额外复制

优化后的代码 #

pub fn initials_optimized(texts: Vec<&str>) -> Vec<String> {
    texts.into_iter().map(|text| {
        let words: Vec<&str> = text.split_whitespace().collect();
        if words.is_empty() {
            return String::new();
        }
        // 计算精确容量：每个单词对应「X. 」是2个字符，最后一个单词不用加空格，总长度=2*单词数-1
        let capacity = 2 * words.len() - 1;
        let mut result = String::with_capacity(capacity);
        
        for (idx, word) in words.iter().enumerate() {
            let first_char = word.chars().next().unwrap();
            result.push(first_char);
            result.push('.');
            // 只有非最后一个单词才加空格
            if idx != words.len() - 1 {
                result.push(' ');
            }
        }
        
        result
    }).collect()
}

迭代器风格的进一步优化 #

如果不想先把单词收集到Vec里，可以用迭代器组合来处理，同时提前统计单词数做预分配：

pub fn initials_iter_optimized(texts: Vec<&str>) -> Vec<String> {
    texts.into_iter().map(|text| {
        let word_count = text.split_whitespace().count();
        if word_count == 0 {
            return String::new();
        }
        
        let mut chars = text.split_whitespace().map(|w| w.chars().next().unwrap());
        let capacity = 2 * word_count - 1;
        let mut result = String::with_capacity(capacity);
        
        // 处理第一个首字母
        if let Some(first) = chars.next() {
            result.push(first);
            result.push('.');
            // 处理剩余首字母，前面加空格
            chars.for_each(|c| {
                result.push(' ');
                result.push(c);
                result.push('.');
            });
        }
        
        result
    }).collect()
}

优化效果说明 #

• 预分配容量让每个结果String只做一次堆内存分配，彻底避免了追加过程中的多次扩容
• 去掉了所有临时字符串的创建，减少了不必要的堆分配
• 不再调用trim_end().to_string()，避免了一次完整的字符串复制

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