Function find_ocurrences

1078-occurrences-after-bigram.rs:1–11

(text: String, first: String, second: String)

pub fn find_ocurrences(text: String, first: String, second: String) -> Vec<String> {
    let strs: Vec<String> = text.split(" ").map(|s|s.to_string()).collect();
    let tail_index = &strs.len() - 2;
    let mut res: Vec<String> = Vec::new();
    for i in 0..tail_index {
        if strs[i] == first && strs[i+1] == second {
            res.push(strs[i+2].clone());
        }
    }
    res
}

fn main() {
    let text = "we will we will rock you".to_string();