Pregunta de entrevista de General Motors (GM)

implement a function in Python