carloscn / structstudy

Leetcode daily trainning by using C/C++/RUST programming.
4 stars 1 forks source link

leetcode1078:Bigram 分词(occurrences-after-bigram) #184

Open carloscn opened 1 year ago

carloscn commented 1 year ago

问题描述

给出第一个词 first 和第二个词 second,考虑在某些文本 text 中可能以 "first second third" 形式出现的情况,其中 second 紧随 first 出现,third 紧随 second 出现。

对于每种这样的情况,将第三个词 "third" 添加到答案中,并返回答案。

 

示例 1:

输入:text = "alice is a good girl she is a good student", first = "a", second = "good" 输出:["girl","student"]

示例 2:

输入:text = "we will we will rock you", first = "we", second = "will" 输出:["we","rock"]  

提示:

1 <= text.length <= 1000 text 由小写英文字母和空格组成 text 中的所有单词之间都由 单个空格字符 分隔 1 <= first.length, second.length <= 10 first 和 second 由小写英文字母组成

来源:力扣(LeetCode) 链接:https://leetcode.cn/problems/occurrences-after-bigram

carloscn commented 1 year ago

问题分析

长度为3的滑动窗口,判断第一是不是first字符,如果是判断第二个是不是second字符,是如果是则取第三个进入的ret_vec中。

pub fn find_ocurrences(text: String, first: String, second: String) -> Vec<String>
{
    let text_vec:Vec<String> = text.split(" ")
                                   .into_iter()
                                   .map(|x| x.to_string())
                                   .collect();
    if text_vec.len() < 3 {
        return vec![];
    }

    let mut ret_vec:Vec<String> = vec![];
    let mut i:usize = 0;

    while i < text_vec.len() - 2 {
        if first == text_vec[i] {
            if second == text_vec[i + 1] {
                ret_vec.push(text_vec[i + 2].clone());
            }
        }
        i += 1;
    }

    return ret_vec;
}
carloscn commented 1 year ago

code

https://review.gerrithub.io/c/carloscn/structstudy/+/552412 https://github.com/carloscn/structstudy/commit/d178f74f2830773c95e48eaa58fead5ba60e2e6d