Решение на Search от Андрей
Резултати
- 15 точки от тестове
- 0 бонус точки
- 15 точки общо
- 5 успешни тест(а)
- 0 неуспешни тест(а)
Код
use std::collections::{HashSet, HashMap};
use std::rc::Rc;
struct WordIterator<'a> {
source: &'a str,
}
impl<'a> WordIterator<'a> {
pub fn new(text: &'a str) -> Self {
WordIterator { source: text }
}
}
impl<'a> Iterator for WordIterator<'a> {
type Item = &'a str;
fn next(&mut self) -> Option<Self::Item> {
let mut on_word = false;
let mut start = None;
let mut end = None;
let mut byte_index = 0;
for c in self.source.chars() {
if !on_word && c.is_alphabetic() {
start = Some(byte_index);
on_word = true;
} else if on_word && !c.is_alphabetic() {
end = Some(byte_index);
on_word = false;
}
if let (Some(start), Some(end)) = (start, end) {
let word = &self.source[start .. end];
self.source = &self.source[end..];
return Some(word);
}
byte_index += c.len_utf8();
}
let start = start?;
let word = &self.source[start..];
self.source = "";
Some(word)
}
}
pub fn extract_words(text: &str) -> Vec<String> {
WordIterator::new(text).map(String::from).collect()
}
pub struct TextIndex {
storage: HashMap<String, HashSet<Rc<String>>>,
}
impl TextIndex {
pub fn new() -> Self {
TextIndex { storage: HashMap::new() }
}
pub fn push(&mut self, text: &str) {
let text = Rc::new(String::from(text));
for word in extract_words(&text) {
let entry = self.storage.entry(word).or_insert(HashSet::new());
entry.insert(text.clone());
}
}
pub fn search(&self, query: &str) -> HashSet<&str> {
let query_words = extract_words(query).into_iter();
let mut results = HashSet::new();
for candidate in query_words.filter_map(|word| self.storage.get(&word)) {
for result in candidate {
results.insert(result);
}
}
results.into_iter().map(Rc::as_ref).map(String::as_str).collect()
}
}
Лог от изпълнението
Compiling solution v0.1.0 (file:///tmp/d20180105-6053-1ubyrsy/solution)
Finished dev [unoptimized + debuginfo] target(s) in 6.17 secs
Running target/debug/deps/solution-3f98bfa5c86a5dd9
running 0 tests
test result: ok. 0 passed; 0 failed; 0 ignored; 0 measured; 0 filtered out
Running target/debug/deps/solution_test-3d9e4ea2eafbbc82
running 5 tests
test solution_test::test_extract_words_basic ... ok
test solution_test::test_extract_words_extra ... ok
test solution_test::test_search_multiple_words ... ok
test solution_test::test_search_special_cases ... ok
test solution_test::test_search_word ... ok
test result: ok. 5 passed; 0 failed; 0 ignored; 0 measured; 0 filtered out
Doc-tests solution
running 0 tests
test result: ok. 0 passed; 0 failed; 0 ignored; 0 measured; 0 filtered out