https://leetcode.com/problems/occurrences-after-bigram Easy

Решение

fun findOcurrences(text: String, first: String, second: String): Array<String> {
    val words = text.split(" ")  // Разделяем текст на отдельные слова
    val result = mutableListOf<String>()  // Создаём список для хранения результата

    for (i in 0..words.size - 3) {  // Перебираем слова, останавливаясь за 2 слова до конца
        if (words[i] == first && words[i + 1] == second) {  // Если текущие слова совпадают с first и second
            result.add(words[i + 2])  // Добавляем следующее слово в результат
        }
    }

    return result.toTypedArray()  // Возвращаем результат в виде массива строк
}