09 Sept 2026•Easy•Go / Python / TypeScript

Occurrences After Bigram

Return every word that immediately follows a specified two-word sequence in text.

Split the text and inspect each consecutive triple, emitting the third word when the first two match the bigram.

Complexity

O(n) time and O(n) token space.

Solution files

Go•occurrences-after-bigram/solution.go

package main

import "strings"

func findOcurrences(text string, first string, second string) []string {
	words := strings.Fields(text)
	result := []string{}
	for index := 2; index < len(words); index++ {
		if words[index-2] == first && words[index-1] == second {
			result = append(result, words[index])
		}
	}
	return result
}

Python•occurrences-after-bigram/solution.py

class Solution:
    def findOcurrences(self, text: str, first: str, second: str) -> list[str]:
        words = text.split()
        return [words[index] for index in range(2, len(words)) if words[index - 2] == first and words[index - 1] == second]

TypeScript•occurrences-after-bigram/solution.ts

function findOcurrences(text: string, first: string, second: string): string[] {
  const words = text.split(" "); const result: string[] = [];
  for (let index = 2; index < words.length; index += 1) if (words[index - 2] === first && words[index - 1] === second) result.push(words[index]);
  return result;
}