题目链接: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 second
comes 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 <= text.length <= 1000
text
consists of space separated words, where each word consists of lowercase English letters.1 <= first.length, second.length <= 10
first
andsecond
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;
}
};