如何将反向域名表示法转换为PascalCase?Rust实现优化咨询
优化Rust中ASCII字符串转驼峰命名的实现
嘿,你的这个把foo.bar.baz转成FooBarBaz的需求,我正好有几个实用的优化思路!先说说你原来代码的问题:你的实现里用了split(".")拆分字符串,然后每个片段都做切片、转String、大写拼接,最后再用fold不断拼接字符串——这会产生大量临时字符串对象,每次拼接都要重新分配内存,时间复杂度是O(n²),确实不够高效。
下面是几个更高效的优化方案,核心思路都是减少临时对象、预先分配内存、直接遍历处理字符:
方案一:预分配容量+遍历字符处理
这个方案是最直接的优化,先计算目标字符串的长度,预先分配足够的内存,然后逐个字符处理:
fn to_camel_case(s: &str) -> String { // 先统计原字符串中点的数量,计算目标字符串的长度 let dot_count = s.chars().filter(|&c| c == '.').count(); // 预先分配足够的内存,避免多次扩容 let mut result = String::with_capacity(s.len() - dot_count); // 标记下一个字符是否需要大写 let mut next_upper = true; for c in s.chars() { match c { '.' => next_upper = true, // 遇到点,标记下一个字符要大写 _ if next_upper => { // 因为输入是ASCII,用to_ascii_uppercase比通用方法更快 result.push(c.to_ascii_uppercase()); next_upper = false; } _ => { // 如果输入可能有大写,这里转小写保证规范;如果确定输入都是小写,直接push(c)即可 result.push(c.to_ascii_lowercase()); } } } result } fn main() { let input = "foo.bar.baz"; let result = to_camel_case(input); println!("{}", result); // 输出 FooBarBaz }
优化点说明:
- 预分配内存:通过统计点的数量,算出目标字符串的长度(原长度 - 点的数量),用
String::with_capacity一次性分配足够的内存,避免String在push过程中多次扩容(每次扩容都会复制现有内容)。 - 直接遍历字符:不需要拆分字符串产生临时的
&str片段,也不用为每个片段创建新的String,每个字符只处理一次,时间复杂度降到O(n)。 - 用ASCII专用方法:因为输入是ASCII字符,
to_ascii_uppercase/to_ascii_lowercase比通用的to_uppercase/to_lowercase更快,不用处理Unicode的复杂情况。
方案二:迭代器链式处理(更简洁的写法)
如果喜欢用迭代器风格,可以用peekable迭代器来处理点后的字符,同样保证高效:
fn to_camel_case(s: &str) -> String { let mut chars = s.chars().peekable(); let dot_count = s.chars().filter(|&c| c == '.').count(); let mut result = String::with_capacity(s.len() - dot_count); while let Some(c) = chars.next() { if c == '.' { // 跳过点,直接取下一个字符转大写加入结果 if let Some(next_c) = chars.next() { result.push(next_c.to_ascii_uppercase()); } } else if result.is_empty() { // 第一个字符直接转大写 result.push(c.to_ascii_uppercase()); } else { result.push(c.to_ascii_lowercase()); } } result }
这个写法更偏向迭代器风格,逻辑同样高效,没有多余的内存分配。
对比原实现的优势
原实现中,每个split出来的片段都要做x[0..1].to_string().to_uppercase() + &x[1..],这会创建多个临时String;fold中的acc + &x又会每次复制现有字符串并拼接,导致大量的内存分配和复制。而优化后的方案全程只维护一个String,每个字符只处理一次,内存分配只有一次,性能提升非常明显,尤其是处理较长的字符串时。
内容的提问来源于stack exchange,提问作者Harald Hoyer
相关产品推荐
相关产品推荐

