Pregunta de entrevista de Lazard

Leetcode medium question, ML and transformers