LeetCode 1078. Bigram 分词

LeetCode题目：

Python代码

class Solution:
    def findOcurrences(self, text: str, first: str, second: str) -> List[str]:
        text = text.split()
        List = []
        #enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列，同时列出数据和数据下标，一般用在 for 循环当中
        #enumerate(sequence, [start=0])
        for index, word in enumerate(text):
            if index > len(text) - 3:
                break
            if word == first and text[index+1] == second:
                List.append(text[index + 2])  
        return List

enumerate方法参考资料

Java代码

import java.util.ArrayList;
import java.util.List;

class Solution {
    public static String[] findOcurrences(String text, String first, String second) {
        String[] s = text.split(" ");
        List<String> res = new ArrayList<>();
        for (int i = 0; i < s.length-2; i++) {
                if (first.equals(s[i]) && second.equals(s[i+1])) {
                    res.add(s[i + 2]);
            }
        }
        String[] result = new String[res.size()];
        res.toArray(result);
        return result;
    }
}