题目
给出第一个词 first 和第二个词 second,考虑在某些文本 text 中可能以 “first second third” 形式出现的情况,其中 second 紧随 first 出现,third 紧随 second 出现。
对于每种这样的情况,将第三个词 “third” 添加到答案中,并返回答案。
解题思路
直接字符串判断就行,但我是懒狗,直接用 split
函数把所有单词分开,再每三个单词进行判断。
代码
class Solution {
public String[] findOcurrences(String text, String first, String second) {
List<String> ans = new ArrayList<>();
String[] splitWords = text.split(" ");
for (int i = 0; i < splitWords.length - 2; i++) {
if (splitWords[i].equals(first) && splitWords[i + 1].equals(second)) {
ans.add(splitWords[i + 2]);
}
}
return ans.toArray(new String[0]);
}
}