我需要编写一个函数来检查两个String
是否在一个字符上以顺序方式不同,即只有diff("aba", "abc") == true
,diff("aab", "cab") == false
,字符串的长度相等)。
我以命令式风格编写代码,但它看起来很丑陋,我想用迭代器以函数式风格编写它。
我想它应该是带有s1.chars()..enumerate()
+ 一些闭包的东西,它检测两个字符串中的一个不同字符。
fn has_one_difference(s1: &String, s2: &String) -> bool {
let mut diff_chars_limit = false;
let mut s1_chars = s1.chars();
let mut s2_chars = s2.chars();
for index in 0..s1.len() {
if s1_chars.nth(index).unwrap() != s2_chars.nth(index).unwrap() {
if diff_chars_limit {
return false
} else {
diff_chars_limit = true
}
}
}
return diff_chars_limit;
}
我收到此错误:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', src/libcore/option.rs:345:21
在字符串迭代的最后一个字符上。
首先,我将你的命令式代码更正为
- 删除对字符迭代器的极低效基于索引的访问,
- 如果
s1
较长但等于s2
,则删除崩溃,并将其替换为代码相反的相同"忽略较长字符串的尾部"行为, - 使用
str
而不是String
因为几乎没有充分的理由将&String
传递给函数,并且 - 修复了一些小的样式问题;特别是向回车符添加分号,但使用尾返回表达式而不返回。这是更惯用的 Rust。
它看起来像这样:
fn has_one_difference(s1: &str, s2: &str) -> bool {
let mut found_one_difference = false;
for (c1, c2) in s1.chars().zip(s2.chars()) {
if c1 != c2 {
if found_one_difference {
return false;
} else {
found_one_difference = true
}
}
}
found_one_difference
}
现在对于功能版本,我将简单地编写一个迭代器,看看我是否可以在它上面调用next()
两次:
fn has_one_difference_functional(s1: &str, s2: &str) -> bool {
// An iterator over different char pairs.
let mut iter = s1.chars().zip(s2.chars())
.filter(|(c1, c2)| c1 != c2);
// First call to next() must succeed (one difference), second must fail.
iter.next().is_some() && iter.next().is_none()
}
这不是完全实用的,但我认为它总体上是简洁性和可读性的最佳组合。一个简单的全功能版本会在组合迭代器上调用count()
并将其与 1 进行比较,但这不会短路,因此效率低于需要。可以使用try_fold
编写更有效的版本,但由于复杂性而失去了可读性,因此我只考虑将其用于has_n_differences
函数。