luoheng
10/11/2019 - 2:22 PM

Bigram


import "strings"

func findOcurrences(text string, first string, second string) []string {
    str := strings.Split(text, " ")
    var res []string
    for i := 0; i < len(str) - 2; i++ {
        if str[i] == first {
            if str[i+1] == second {
                res = append(res, str[i+2])
            }
        }
    }
    return res
}