LeetCode1078 Bigram 分词

1078. Bigram 分词

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

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

运用JAVA的split函数可以有效降低难度，当然也可以遍历通过寻找" "来分割

class Solution {
    public String[] findOcurrences(String text, String first, String second) {
        //split函数，根据空格将text分割成多个字符串
        String[] words = text.split(" ");
        List<String> list = new ArrayList<String>();
        //逐个匹配，如果能确保匹配前两个就将第三个加入list
        for (int i = 2; i < words.length; i++) {
            if (words[i - 2].equals(first) && words[i - 1].equals(second)) {
                list.add(words[i]);
            }
        }
        int size = list.size();
        //list转数组
        String[] ans = new String[size];
        for (int i = 0; i < size; i++) {
            ans[i] = list.get(i);
        }
        return ans;
    }
}