1078. Occurrences After Bigram
Given words first
and second
, consider occurrences in some text
of the form "first second third
", where second
comes immediately after first
, and third
comes immediately after second
.
For each such occurrence, add "third
" to the answer, and return the answer.
Example 1:
Input: text = "alice is a good girl she is a good student", first = "a", second = "good"
Output: ["girl","student"]
Example 2:
Input: text = "we will we will rock you", first = "we", second = "will"
Output: ["we","rock"]
Note:
1 <= text.length <= 1000
text
consists of space separated words, where each word consists of lowercase English letters.1 <= first.length, second.length <= 10
first
andsecond
consist of lowercase English letters.
# @lc code=start
using LeetCode
function occurrences_after_bigram(text::String, first::String, second::String)
words = split(text)
res = String[]
for i in 1:length(words)-2
if words[i] == first && words[i+1] == second
push!(res, words[i+2])
end
end
return res
end
# @lc code=end
occurrences_after_bigram (generic function with 1 method)
This page was generated using DemoCards.jl and Literate.jl.