题目要求:
给出第一个词 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”]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/occurrences-after-bigram
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
问题分析:
本道题主要是对单词进行比较,所以我们可以将一个单词看作一个元素,用空格分开,将每个元素放入一个数组中,与first和second分别进行比较,如果前两个相等,就返回第三个的单词,多个返回可以使用ArrayList数组进行存储,因为不知道容量大小,所以ArrayList是最适合存放结果的,最后使用ArrayList.toArray( new T[0]) 方法进行转换成数组即可。
解题代码:
class Solution {
public String[] findOcurrences(String text, String first, String second) {
String[] strArr = text.split(" ");
List<String> list = new ArrayList<String>();
for(int i = 2;i<strArr.length;i++){
if(first.equals(strArr[i-2])&&second.equals(strArr[i-1])){
list.add(strArr[i]);
}
}
return list.toArray(new String[0]);
}
}