Bigram 分词

/*
 * 思路：
 * 1. 将输入的 text 分割成单词数组。
 * 2. 使用 Java Stream API 处理数组并查找连续的 'first' 和 'second'。
 * 3. 将符合条件的 'third' 单词收集到结果列表中。
 */

import java.util.ArrayList;
import java.util.List;
import java.util.stream.IntStream;

public class Solution {
    public List<String> findOcurrences(String text, String first, String second) {
        String[] words = text.split(" ");
        List<String> result = new ArrayList<>();
        IntStream.range(0, words.length - 2)
                .filter(i -> words[i].equals(first) && words[i + 1].equals(second))
                .forEach(i -> result.add(words[i + 2]));
        return result;
    }
}