【刷leetcode，拿Offer-025】1078. Occurrences After Bigram

题目链接：https://leetcode.com/problems/occurrences-after-bigram/

题面：

1078. Occurrences After Bigram

Easy

3138FavoriteShare

Given words first and second, consider occurrences in some text of the form "first second third", where secondcomes immediately after first, and third comes immediately after second.

For each such occurrence, add "third" to the answer, and return the answer.

 

Example 1:

Input: text = "alice is a good girl she is a good student", first = "a", second = "good"
Output: ["girl","student"]

Example 2:

Input: text = "we will we will rock you", first = "we", second = "will"
Output: ["we","rock"]

 

Note:

1. 1 <= text.length <= 1000
2. text consists of space separated words, where each word consists of lowercase English letters.
3. 1 <= first.length, second.length <= 10
4. first and second consist of lowercase English letters.

题意：输出连着的两个单词词组的后一个单词。

代码：

class Solution {
public:
    vector<string> findOcurrences(string text, string first, string second) {
        vector <string> ans,tmp;
        string str;
        text+=" ";
        for(int i=0;i<text.length();i++){
            if(text[i]==' ')
            {
                tmp.push_back(str);
                str="";
            }
            else
                str+=text[i];
        }
        for(int i=2;i<tmp.size();i++){
            if(tmp[i-2]==first&&tmp[i-1]==second)
                ans.push_back(tmp[i]);
        }
        return ans;
    }
};