lib.rs

use std::collections::{HashMap, HashSet};


pub fn word_count(input: &str) -> HashMap<String, u32> {
    let mut map: HashMap<String, u32> = HashMap::new();
    let words: Vec<&str> = input.split_whitespace().collect();
    let set: HashSet<&str> = words.clone().into_iter().collect::<HashSet<&str>>();

    for word in set {
        let count = words.iter().filter(|&w| *w == word).count() as u32;
        map.insert(word.to_string(), count);
    }
    map
}